tvm_tirx.backend.cuda.op.atomic_add

Contents

tvm_tirx.backend.cuda.op.atomic_add#

tvm_tirx.backend.cuda.op.atomic_add(res_addr, value, *, ty=None, loc=UNKNOWN_LOC)#

TVM intrinsic to call cuda atomic add instruction

Parameters:
  • res_addr (Expr) – The result address.

  • value (Expr) – The value to add.

  • loc (Location)

Returns:

call (Expr) – The call expression.