cupyx.jit.atomic_dec

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

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

参数:

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

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

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

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

另请参阅

Numba 中对应的原子函数