torch-mlir/lib/CAPI
Stella Laurenzo 9e52f6235b More progress on PyTorch acap device capture.
* Now gets far enough to capture batch_norm.
* Has some issues still with in-place ops.
* Can materialize constants.
* Includes an upgrade to PyTorch nightly, which has important bug fixes for fallback and boxed kernel dispatch.
* Fixes #78, #79, #80.
* Will do more testing in a follow-up once further bugs are fixed that facilitate getting at the other features.
2020-10-15 21:43:21 -07:00
..
CMakeLists.txt Start reworking towards a shared library build. 2020-10-09 16:02:58 -07:00
InitLLVM.cpp Start reworking towards a shared library build. 2020-10-09 16:02:58 -07:00
Registration.cpp Add public API for constructing a module/function to capture PyTorch ops. 2020-09-29 14:23:22 -07:00
Types.cpp More progress on PyTorch acap device capture. 2020-10-15 21:43:21 -07:00