triton.language.store

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

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

value is implicitly broadcast to pointer.shape and typecast to pointer.dtype.element_ty.

Parameters

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

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

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

  • builder (triton.ir.builder, optional from within JIT'ed functions) – IR builder to generate code into