博客 / 詳情

返回

【Triton 教程】triton.language.advance

Triton 是一種用於並行編程的語言和編譯器。它旨在提供一個基於 Python 的編程環境,以高效編寫自定義 DNN 計算內核,並能夠在現代 GPU 硬件上以最大吞吐量運行。

更多 Triton 中文文檔可訪問 →triton.hyper.ai/

triton.language.advance(base, offsets)

推進 1 個塊指針。

參數

  • base - 要推進的塊指針。
  • offsets - 要推進的偏移量,是一個按維度劃分的元組。

這個函數也可作為 tensor 的成員函數調用,使用 x.advance(...) 的方式而不是 advance(x, ...)

user avatar
0 位用戶收藏了這個故事!

發佈 評論

Some HTML is okay.