triton.language.load

triton.language.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.

mask and other are implicitly broadcast to pointer.shape.

other is implicitly typecast to pointer.dtype.element_ty.

Parameters

• pointer (Block of dtype=triton.PointerDType) – Pointers to the data to be loaded.

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

• other (Block, optional) – if mask[idx] is false, return other[idx]

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