torch-mlir/python
Yi Zhang 05c4dd8e39 Add convertScalarToDtype helper.
This is to facilitate scalar type conversion in the TorchToLinalg. As
part of adding the helper, this PR also:
- Updated `AtenAddTensorOp`, `AtenSubTensorOp` to use the helpers to
support more type variants.
- Added e2e type promotion testing.
- Added i32 memref return/arg type to support e2e testing.
2021-11-08 17:50:52 -05:00
..
test Make error reporting a bit better. 2021-10-28 13:20:16 -07:00
torch_mlir/dialects Add log2 2021-11-08 16:19:59 -05:00
torch_mlir_e2e_test Add convertScalarToDtype helper. 2021-11-08 17:50:52 -05:00
CMakeLists.txt Enable python extensions when building out of tree (#363) 2021-10-27 17:04:12 -07:00
TorchMLIRModule.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00