mirror of https://github.com/llvm/torch-mlir
4dd213b042
- Add/Extend Torch to TOSA legalization for the following ops: + Add aten.threshold_backward + Fix aten.threshold + Re-implement aten.broadcast_to using tosa.reshape and tosa.tile + Add support for rank 0 index for aten.index_select + Fix aten.index_put.hacked_twin + Add aten.uniform + Add aten.logical_and - Update xfail_sets.py with new e2e results - Add LIT tests to basic.mlir for newly added ops Change-Id: I8910564a049d18293284fe2e55e82bc1d2cf10e3 Signed-off-by: Justin Ngo <justin.ngo@arm.com> |
||
---|---|---|
.. | ||
basic.mlir | ||
cast_fp32_to_fp16.mlir | ||
conv2d_transpose.mlir | ||
torch-backend-to-tosa-backend-pipeline.mlir |