【Triton 教程】triton_language.program_id
💡
原文中文,约300字,阅读约需1分钟。
📝
内容提要
Triton是一种基于Python的并行编程语言和编译器,旨在高效编写自定义深度神经网络计算内核,并在现代GPU上实现高吞吐量运行。
🎯
关键要点
- Triton是一种用于并行编程的语言和编译器。
- Triton提供基于Python的编程环境。
- Triton旨在高效编写自定义深度神经网络计算内核。
- Triton能够在现代GPU上实现高吞吐量运行。
- 提供更多Triton中文文档的链接。
- triton.language.program_id(axis)函数用于返回当前程序实例的ID。
➡️