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/c10_dispatch
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
..
CMakeLists.txt Start reworking towards a shared library build. 2020-10-09 16:02:58 -07:00
acap_dispatch.cpp More progress on PyTorch acap device capture. 2020-10-15 21:43:21 -07:00
acap_dispatch.h More progress on PyTorch acap device capture. 2020-10-15 21:43:21 -07:00
func_builder.cpp More progress on PyTorch acap device capture. 2020-10-15 21:43:21 -07:00
func_builder.h More progress on PyTorch acap device capture. 2020-10-15 21:43:21 -07:00
module_builder.cpp Create public API for torch_mlir python code. 2020-10-13 16:36:49 -07:00
module_builder.h Create public API for torch_mlir python code. 2020-10-13 16:36:49 -07:00
python_bindings.cpp Implement torch.kernel_call capture. 2020-10-06 21:54:28 -07:00