torch-mlir/test/Conversion/TorchToTosa
TatWai Chong ed4ecb072f
[tosa] support lowering basic torch binary ops with mixed dtypes (#2122)
Lowering torch operations that allow different compatible data types
in its operands to tosa end up generating invalid tosa IR with mixed
data types. In tosa spec, certain operations (generally element-wise
operations) require all operands to have the same data type.

Add wrapper functions for those element-wise tosa ops to perform op
creation with type conversion if necessary.
2023-05-18 17:12:18 -07:00
..
basic.mlir Integrate LLVM 2023-05-09 10:14:27 -07:00
torch-backend-to-tosa-backend-pipeline.mlir [tosa] support lowering basic torch binary ops with mixed dtypes (#2122) 2023-05-18 17:12:18 -07:00