llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
299 Commits
47 Branches
1195 Tags
63 MiB
Commit Graph

3 Commits (81119aa0a10d1ee759ac0641e406b89a793a7a48)

Author SHA1 Message Date
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
Stella Laurenzo ad3ddb9edb Implement torch.kernel_call capture. 
* Had to stop short of modifying the function return signature because of a missing C-API upstream.
* Committing here is good enough for a test and will resolve the various TODOs about upstream APIs next.
 2020-10-06 21:54:28 -07:00
Stella Laurenzo e5433e314f Add capture function arguments. 
* Adds at::Tensor -> MlirValue tracking.
* Adds conversions for tensor and scalar types to MLIR types.
* Adds npcomp C APIs for constructing custom types.
* Reworks pybind include so as to get Torch pybind helpers (needed to pass at::Tensor type from Python->C++).
 2020-10-01 18:59:58 -07:00