triton.language.make_block_ptr

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

返回父张量中一个块的指针

参数:

• base – 父张量的基本指针

• shape – 父张量的形状

• strides – 父张量的跨度

• offsets – 块的偏移量

• block_shape – 块的形状

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