triton.language.atomic_add

triton.language.atomic_add(pointer, val, mask=None)

Performs an atomic add at the memory location specified by pointer.

Return the data stored at pointer before the atomic operation.

Parameters:

  • pointer (Block of dtype=triton.PointerDType) – The memory locations to compare-and-swap.

  • cmp (Block of dtype=`pointer.dtype.element_ty`) – The values expected to be found in the atomic object

  • val (Block of dtype=`pointer.dtype.element_ty`) – The values to copy in case the expected value matches the contained value.