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. |
||
---|---|---|
.. | ||
test | ||
torch_mlir/dialects | ||
torch_mlir_e2e_test | ||
CMakeLists.txt | ||
TorchMLIRModule.cpp |