triton.language.make_block_ptr

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

返回指向父张量中某个数据块(block)的指针

参数:
  • base – 指向父张量的基指针

  • shape – 父张量的形状(shape)

  • strides – 父张量的步长(strides)

  • offsets – 指向该数据块的偏移量

  • block_shape – 该数据块的形状

  • order – 原始数据格式的排列顺序