torch-mlir/python/torch_mlir_e2e_test/torchscript
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
..
configs Bring some more order to the e2e error reporting situation. 2021-10-08 13:00:12 -07:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
__init__.py Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
annotations.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
framework.py Add min/max/clamp support. 2021-10-27 13:29:21 -07:00
registry.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
reporting.py Add convertScalarToDtype helper. 2021-11-08 17:50:52 -05:00