triton.language.atomic_add

triton.language.atomic_add(pointer, val, mask=None, sem=None, scope=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 operate on

  • val (Block of dtype=pointer.dtype.element_ty) – The values with which to perform the atomic operation

  • sem (str) – Memory semantics to use (“ACQUIRE_RELEASE” (default), “ACQUIRE”, “RELEASE”, or “RELAXED”)

  • scope (str) – Scope of threads that observe synchronizing effect of the atomic operation (“GPU” (default), “CTA”, or “SYSTEM”)

This function can also be called as a member function on tensor, as x.atomic_add(...) instead of atomic_add(x, ...).