【Triton 教程】triton.language.advance

28 阅读1分钟

Triton 是一种用于并行编程的语言和编译器。它旨在提供一个基于 Python 的编程环境,以高效编写自定义 DNN 计算内核,并能够在现代 GPU 硬件上以最大吞吐量运行。

更多 Triton 中文文档可访问 →triton.hyper.ai/

推进 1 个块指针。

参数**:**

  • base - 要推进的块指针。
  • offsets - 要推进的偏移量,是一个按维度划分的元组。

这个函数也可作为 tensor 的成员函数调用,使用 x.advance(...) 的方式而不是 advance(x, ...)