mirror of https://github.com/llvm/torch-mlir
05c4dd8e39
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. |
||
---|---|---|
.. | ||
configs | ||
CMakeLists.txt | ||
__init__.py | ||
annotations.py | ||
framework.py | ||
registry.py | ||
reporting.py |