mirror of https://github.com/llvm/torch-mlir
0f80e75c2e
According to the [official TOSA spec](https://www.mlplatform.org/tosa/tosa_spec.html#_cast), `tosa.cast` allows a cast from `fp32` to `fp16`. We were not previously accounting for this in the `TorchToTosa` lowering. Also did a tiny bit of cleanup in the code to make it easier to spot which conversions are currently allowed. --------- Co-authored-by: Srinath Avadhanula <srinath.avadhanula@getcruise.com> |
||
---|---|---|
.. | ||
basic.mlir | ||
cast_fp32_to_fp16.mlir | ||
conv2d_transpose.mlir | ||
torch-backend-to-tosa-backend-pipeline.mlir |