pytorch
3654552b - add deterministic impl for scatter and scatter_reduction sum/mean mode (#98060)

Commit
1 year ago
add deterministic impl for scatter and scatter_reduction sum/mean mode (#98060) using the existing deterministic implementation via `index_put` which has a deterministic implementation based on sorting indices. With the `accumulate` arg in `index_put`, this can work for both scatter and scatter_reduce with sum/mean reduction mode. Pull Request resolved: https://github.com/pytorch/pytorch/pull/98060 Approved by: https://github.com/mikaylagawarecki
Author
Committer
Parents
Loading