mirror of https://github.com/llvm/torch-mlir
5009cbf55c
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> |
||
---|---|---|
.. | ||
test | ||
torch_mlir/dialects | ||
torch_mlir_e2e_test | ||
CMakeLists.txt | ||
TorchMLIRModule.cpp |