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
Ramiro Leal-Cavazos 9ce62473f9
Add static type information support to `aten.bmm` (#636) 
This commit adds static type information support to `aten.bmm`. This
is needed for the forward pass of Bert training.
 		2022-03-03 13:01:17 -08:00
..
CAPI [Torch][Linalg] Add basic support for RNG 2022-01-31 18:56:42 -05:00
Conversion [LINALG] Add torch-to-linalg lowering for `TensorStaticInfoCastOp` (#634) 2022-03-02 13:35:26 -08:00
Dialect Add static type information support to `aten.bmm` (#636) 2022-03-03 13:01:17 -08:00
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