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. |
||
---|---|---|
.. | ||
e2e_testing | ||
examples | ||
python | ||
test | ||
tools | ||
CMakeLists.txt |