triton.store

triton.store(pointer, value, mask=None, builder=None)

Stores value block of elements in memory, element-wise, at the memory locations specified by pointer.

Parameters

• pointer (Block of triton.pointer) – The memory locations where the elements of value are stored.

• value (Block of triton.value) – The block of elements to be stored.

• mask (Block of triton.bool, optional) – If mask[idx] is false, do not store value[idx] at pointer[idx].

• builder (triton.ir.builder) – IR builder to generate code into, optional from within @triton.jit functions