cupyx.jit.atomic_max#

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

调用 atomicMax 函数对 array[index] 进行原子操作。详细解释请参考 原子函数 (Atomic Functions)。

参数:

  • array – 用于索引的 cupy.ndarray。

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

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

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

另请参阅

Numba 对应的原子函数