triton.load

triton.load(pointer, mask=None, other=None, builder=None)

Return a block of data whose values are, elementwise, loaded from memory at location defined by pointer.

Parameters

• pointer (Block of triton.pointer) – Pointer to the data to be loaded.

• mask (Block of triton.bool, optional) – if mask[idx] is false, do not load the data at pointer[idx].

• other (Block of triton.value, optional) – if mask[idx] is false, return other[idx] instead of ‘pointer[idx]`

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