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
,表示要交换到的值。
另请参阅