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

54 Commits (5f1b2ba3232181551cbe3c31b72c409de0d64bf4)

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 3d74337be0 Add a torch.kernel_call op and associated predicates. 2020-09-29 15:10:38 -07:00
Stella Laurenzo 2c9ca79c89 Add boilerplate for Torch dialect. 2020-09-28 15:26:17 -07:00