cupyx.jit.atomic_cas#

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

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

参数:
  • array – 要对其进行索引的 cupy.ndarray

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

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

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