mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 00:20:18 +01:00
9ca080db87
That fixes `index_put(..., accumulate=True)` for all dtypes int64 operation is not really atomic, but eventually consistent from the `index_put_accumulate` kernel point of view: i.e. by the end of the operation results in the global memory are indeed accumulation of the operands at given indices Pull Request resolved: https://github.com/pytorch/pytorch/pull/158179 Approved by: https://github.com/dcci, https://github.com/Skylion007 ghstack dependencies: #158064, #158178 |
||
|---|---|---|
| .. | ||
| atomic.h | ||
| common.h | ||
| expm1f.h | ||
| indexing.h | ||
| random.h | ||
| reduction_utils.h | ||
| special_math.h | ||
| utils.h | ||