triton.language.make_block_ptr

triton.language.make_block_ptr(base: tensor, shape, strides, offsets, block_shape, order)

Returns a pointer to a block in a parent tensor

Parameters:
  • base – The base pointer to the parent tensor

  • shape – The shape of the parent tensor

  • strides – The strides of the parent tensor

  • offsets – The offsets to the block

  • block_shape – The shape of the block

  • order – The order of the original data format