cupyx.jit.atomic_exch#

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

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

参数:
  • array – 一个用于索引的 cupy.ndarray

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

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

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