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 – Represent the value to use for the specified operation. For the case of atomic_cas, this is the value for array[index] to compare with.

  • alt_value – Only used in atomic_cas to represent the value to swap to.

参见

Numba 的相应原子函数