cupyx.jit.atomic_add#

cupyx.jit.atomic_add(array, index, value, alt_value=None) = <cupyx.jit function>#

调用 atomicAdd 函数对 array[index] 进行原子操作。有关详细说明，请参阅 原子函数。

参数:

  • array – 要索引的 cupy.ndarray。

  • index – 有效索引，可计算对应数组元素 array[index] 的地址。

  • value – 表示用于指定操作的值。对于 atomic_cas 的情况，这是与 array[index] 进行比较的值。

  • alt_value – 仅用于 atomic_cas 中，表示要交换到的值。

另请参阅

Numba 的对应原子函数