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. |
||
---|---|---|
.. | ||
linalg_on_tensors_backends | ||
torchscript | ||
tosa_backends | ||
CMakeLists.txt | ||
__init__.py | ||
utils.py |