torch-mlir

History

Ian Wood e88faf08ff Create scatter op with unique indicies (#3853 ) For the op `index_put_`, if accumulate == false, the behavior is undefined if the indicies aren't unique (https://pytorch.org/docs/stable/generated/torch.Tensor.index_put_.html). So, when converting `AtenIndexPutHackedTwinOp` to a TMTensor scatter op, mark the indices as unique if when `accumulate == false`. This should have no functional effect (unless users are relying on UB) and assuming unique indices has the benefit of unlocking better optimizations in further compiler stages. Signed-off-by: Ian Wood <ianwood2024@u.northwestern.edu>	2024-11-05 12:48:34 -08:00
..
CMakeLists.txt	Re-organize project structure to separate PyTorch dependencies from core project. (#2542 )	2023-11-02 19:45:55 -07:00
TorchToTMTensor.cpp	Create scatter op with unique indicies (#3853 )	2024-11-05 12:48:34 -08:00

Create scatter op with unique indicies (#3853 )

For the op `index_put_`, if accumulate == false, the behavior is
undefined if the indicies aren't unique
(https://pytorch.org/docs/stable/generated/torch.Tensor.index_put_.html).
So, when converting `AtenIndexPutHackedTwinOp` to a TMTensor scatter op,
mark the indices as unique if when `accumulate == false`.

This should have no functional effect (unless users are relying on UB)
and assuming unique indices has the benefit of unlocking better
optimizations in further compiler stages.

Signed-off-by: Ian Wood <ianwood2024@u.northwestern.edu>

2024-11-05 12:48:34 -08:00

CMakeLists.txt

Re-organize project structure to separate PyTorch dependencies from core project. (#2542 )

2023-11-02 19:45:55 -07:00

TorchToTMTensor.cpp

Create scatter op with unique indicies (#3853 )

2024-11-05 12:48:34 -08:00