torch-mlir/lib
Ramiro Leal-Cavazos e966112c8d
Add final cast to TorchToLinalg conversions missing it (#692)
In order to make sure that the TorchToLinalg conversions leave the
graph in a valid state, the final result of the conversion has to be
casted to the result type of the op. This commit adds this cast to ops
that did not have it.
2022-03-23 13:52:32 -07:00
..
CAPI [Torch][Linalg] Add basic support for RNG 2022-01-31 18:56:42 -05:00
Conversion Add final cast to TorchToLinalg conversions missing it (#692) 2022-03-23 13:52:32 -07:00
Dialect Add non-default dtype support for a few elementwise math ops. (#687) 2022-03-23 13:35:43 -07:00
RefBackend Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04: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