cupyx.jit.atomic_cas#

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

Calls the atomicCAS function to operate atomically on array[index]. Please refer to Atomic Functions for detailed explanation.

Parameters:
  • array – A cupy.ndarray to index over.

  • index – A valid index such that the address to the corresponding array element array[index] can be computed.

  • value – Represent the value to use for the specified operation. For the case of atomic_cas, this is the value for array[index] to compare with.

  • alt_value – Only used in atomic_cas to represent the value to swap to.