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

3 Commits (572d198b683bfcfd57e75a26e943340ccba03719)

Author SHA1 Message Date
Sean Silva b2077738ca Bump llvm-project to 444822d77a7fea28aa49edf24533c987efa1b2ee 
Fixes:
- renames StandardTypes -> BuiltinTypes
- std.extract_element -> tensor.extract
 2020-12-11 14:43:38 -08:00
Stella Laurenzo 510f226df2 Expose signature metadata to ops and implement ATenRecognizeKernelsPass pass. 
* Two op interfaces, one for querying instance metadata and one for getting static data needed to construct an op from a generic form.
* For torch.generic_kernel ops, metadata is splatted in during capture from Torch (it comes from the op registry, which will work for either device capture or graph import).
* Moved the 'add' out of the generated set so I can experiment on it. It implements the TorchBuildableKernelOpInterface interface which provides its metadata.
* The ATenRecognizeKernelsPass pass generically lowers from a torch.generic_kernel to recognized ops that implement the TorchBuildableKernelOpInterface, handling the various types of transformations that we allow at this stage.
 2020-10-26 20:31:45 -07:00
Stella Laurenzo 9618c2dbf7 NFC: Re-organize ATen directory structure and fix warnings. 
* Still some more work to do on the Transforms tree to bring it in line with the others (will do that as I add things).
 2020-10-22 14:13:26 -07:00