llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/frontends/pytorch/csrc
History
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
..
c10_dispatch More progress on PyTorch acap device capture. 2020-10-15 21:43:21 -07:00
type_dispatch Add capture function arguments. 2020-10-01 18:59:58 -07:00
CMakeLists.txt Start reworking towards a shared library build. 2020-10-09 16:02:58 -07:00
init_python_bindings.cpp Add public API for constructing a module/function to capture PyTorch ops. 2020-09-29 14:23:22 -07:00
init_python_bindings.h Add capture function arguments. 2020-10-01 18:59:58 -07:00
pybind.h Add capture function arguments. 2020-10-01 18:59:58 -07:00