triton.language.load

triton.language.load(pointer, mask=None, other=None, cache_modifier='')

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]

• cache_modifier – changes cache option in nvidia ptx

‘type cache_modifier: str, optional