llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib
History
Vivek Khandelwal 1a2a9e066f [MLIR][TORCH] Add TorchToTMTensor pass 
This pass is added to lower ops, which can not be lowered
via the TorchToLinalg pass, such as `torch.bincount` op.
This pass also uses torch-mlir's TMTensor Dialect to lower the
complex ops.

Also add torch.bincount op lowering with the help of TMTensor dialect

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
 		2022-03-08 22:52:34 +05:30
..
CAPI [Torch][Linalg] Add basic support for RNG 2022-01-31 18:56:42 -05:00
Conversion [MLIR][TORCH] Add TorchToTMTensor pass 2022-03-08 22:52:34 +05:30
Dialect [MLIR][TORCH] Add TorchToTMTensor pass 2022-03-08 22:52:34 +05:30
RefBackend LLVM bump 2022-02-16 15:28:13 -05:00
CMakeLists.txt Add TMTensor dialect to torch-mlir 2022-02-15 16:45:38 -05:00
InitAll.cpp Add TMTensor dialect to torch-mlir 2022-02-15 16:45:38 -05:00