triton.language.advance

triton.language.advance(base, offsets)

推进块指针

Parameters:

  • base – 要推进的块指针

  • offsets – 要前进的偏移量，按维度的元组

该函数也可以作为成员函数在tensor上调用， 使用x.advance(...)而非 advance(x, ...)。