torch-mlir/python
Prashant Kumar 1d5b5a89e8 [LINALG] Add torch.layout information
torch.layout information has been added.
2022-04-07 20:47:49 +05:30
..
test Make error reporting a bit better. 2021-10-28 13:20:16 -07:00
torch_mlir Add link to forum post describing `__torch_dispatch__` 2022-04-01 10:10:43 -07:00
torch_mlir_e2e_test [LINALG] Add torch.layout information 2022-04-07 20:47:49 +05:30
CMakeLists.txt This PR implements an eager mode backend for PyTorch through the torch-mlir framework. This is accomplished by overriding the `__torch_dispatch__` class method on wrapper subclass `TorchMLIRTensor(torch.Tensor)`. 2022-03-22 14:42:57 -07:00
TorchMLIRModule.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00