mirror of https://github.com/llvm/torch-mlir
b75d0e3f8b
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. |
||
---|---|---|
.. | ||
main.py | ||
xfail_sets.py |