torch-mlir/include/npcomp/Dialect/Torch/IR
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 Add boilerplate for Torch dialect. 2020-09-28 15:26:17 -07:00
TorchBase.td More progress on PyTorch acap device capture. 2020-10-15 21:43:21 -07:00
TorchDialect.h Add boilerplate for Torch dialect. 2020-09-28 15:26:17 -07:00
TorchOps.h Add boilerplate for Torch dialect. 2020-09-28 15:26:17 -07:00
TorchOps.td Implement torch.kernel_call capture. 2020-10-06 21:54:28 -07:00