cupyx.jit.atomic_inc#

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

调用 atomicInc 函数对 array[index] 执行原子操作。详细说明请参阅原子函数。

参数:

  • array – 要进行索引的 cupy.ndarray。

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

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

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

另请参阅

Numba 中对应的原子函数