torch-mlir/projects/pt1/e2e_testing
Jiawei Wu b75d0e3f8b
[stablehlo] fix: enhance torch's index-like op lowering to stablehlo's gather/scatter (#3829)
In torch.index_put like ops, `values` is only required to be
broadcastable to `input[indices]`, rather than exact dimension match.
This patch fixes the problem by add additional
stablehlo.dynamic_broadcast_in_dim before creating stablehlo.scatter op.
BTW, this patch also enhance the `getBroadcastResultShape` utility in
hlo namespace.
2024-11-05 19:15:11 +08:00
..
main.py [TOSA] Add Torch to Tosa Legalization for torch.tril (#3678) 2024-09-05 11:27:29 -07:00
xfail_sets.py [stablehlo] fix: enhance torch's index-like op lowering to stablehlo's gather/scatter (#3829) 2024-11-05 19:15:11 +08:00