torch-mlir/lib/Dialect/Torch
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
..
IR Add aten.ne.bool to make CI pass 2021-10-21 14:45:41 -04:00
Transforms Add convertScalarToDtype helper. 2021-11-08 17:50:52 -05:00
Utils Add type promotion code to refine types. 2021-10-29 11:17:39 -04:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00