tvm_tirx.backend.cuda.op.atomic_cas

Contents

tvm_tirx.backend.cuda.op.atomic_cas#

tvm_tirx.backend.cuda.op.atomic_cas(ptr, old_val, new_val, *, ty=None, loc=UNKNOWN_LOC)#

TVM intrinsic to call cuda atomic cas instruction

Parameters:

  • ptr (Expr) – The pointer to the memory location.

  • old_val (Expr) – The old value.

  • new_val (Expr) – The new value.

  • loc (Location)

Returns:

call (Expr) – The call expression.

Contents