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/Conversion
History
Prashant Kumar 5009cbf55c Add lowering of aten.matmul op. 
Lowering of `aten.matmul` op is added from torch to linalg dialect.
The different cases correspond to
https://pytorch.org/docs/stable/generated/torch.matmul.html.
TODO: Broadcasting in case of batch-matmul is yet to be taken care of.

Signed-off-by: Prashant Kumar <prashant@nod-labs.com>
 		2021-10-26 12:45:09 -04:00
..
TorchToLinalg Add lowering of aten.matmul op. 2021-10-26 12:45:09 -04:00
TorchToSCF Add a basic TOSA E2E backend. 2021-10-08 09:59:45 -07:00
TorchToStd Update llvm-project to a54f4eae0e1d0ef5adccdcf9f6c2b518dc1101aa 2021-10-18 13:36:42 -04:00
TorchToTosa Add a basic TOSA E2E backend. 2021-10-08 09:59:45 -07:00
CMakeLists.txt Add a basic TOSA E2E backend. 2021-10-08 09:59:45 -07:00
PassDetail.h Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
Passes.cpp Add a basic TOSA E2E backend. 2021-10-08 09:59:45 -07:00