【Triton 教程】triton_language.interleave
原文中文,约300字,阅读约需1分钟。
📝
内容提要
Triton 是一种基于 Python 的并行编程语言和编译器,旨在高效编写自定义深度神经网络计算内核,并在现代 GPU 上实现最佳性能。
🎯
关键要点
-
Triton 是一种用于并行编程的语言和编译器。
-
Triton 提供基于 Python 的编程环境。
-
Triton 旨在高效编写自定义 DNN 计算内核。
-
Triton 能够在现代 GPU 硬件上以最大吞吐量运行。
-
Triton 提供中文文档链接。
-
triton.language.interleave 函数用于沿最后一个维度交错两个张量的值。
-
交错的两个张量必须具有相同的形状。
🏷️