warp.tile_atomic_add#

warp.tile_atomic_add(
a: Array[Any],
t: Tile[Any, tuple[int, ...]],
offset: tuple[int, ...],
bounds_check: bool,
) → Tile[Any, tuple[int, ...]]#

  • Kernel

  • Differentiable

Atomically add a tile onto the array a.

Each element is updated atomically.

Parameters:

  • a – Array in global memory, should have the same dtype as the input tile

  • t – Source tile to add to the destination array

  • offset – Offset in the destination array (optional)

  • bounds_check – Needed for unaligned tiles, but can disable for memory-aligned tiles for faster write times

Returns:

A tile with the same dimensions and data type as the source tile, holding the original value of the destination elements.

warp.tile_atomic_add(
a: Array[Any],
t: Tile[Any, tuple[int, ...]],
offset: int32,
bounds_check: bool,
) → Tile[Any, tuple[int, ...]]

  • Kernel

  • Differentiable

Atomically add a tile onto the array a.