【Triton 教程】triton_language.fma

【Triton 教程】triton_language.fma

💡 原文中文,约300字,阅读约需1分钟。
📝

内容提要

Triton是一种基于Python的并行编程语言和编译器,用于高效编写自定义DNN计算内核,并在现代GPU上实现最大吞吐量。其triton.language.fma(x, y, z)函数执行逐元素融合乘加运算。

🏷️

标签

➡️

继续阅读