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

4 Commits (029815152e6a072e3a7f8d817accaae18d6877e1)

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