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

4 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 3ccc2214a7 Set PyTorch captured function return type. 
* Resolves various TODOs that required an LLVM change/bump.
* Bumps LLVM to 4aa217160e5f06a96c6effc4950c3b402374de58
 2020-10-07 10:14:34 -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 b5f010284f Add boilerplate to do device capture (pytorch 1.6). 
* Uses the new dispatcher API.
* Just prints to the console for the moment when an op is captured.
* Executes the op through the existing implementation.
 2020-09-28 10:30:54 -07:00