triton.language.make_block_ptr

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

返回指向父张量中某块的指针

Parameters:
  • base – 指向父张量的基础指针

  • shape - 父张量的形状

  • strides – 父张量的步长

  • offsets – 块的偏移量

  • block_shape – 块的形状

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