triton.language.load

triton.language.load(pointer, mask=None, other=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]