【Triton 教程】triton.heuristics

💡 原文中文,约600字,阅读约需2分钟。
📝

内容提要

Triton 是一种基于 Python 的并行编程语言和编译器,旨在高效编写 DNN 计算内核并在现代 GPU 上运行。它提供 @triton.heuristics 装饰器,用于在自动调优不适用时指定元参数值的计算方法。

🎯

关键要点

  • Triton 是一种用于并行编程的语言和编译器,基于 Python。
  • Triton 旨在高效编写自定义 DNN 计算内核,并在现代 GPU 上运行。
  • 提供 @triton.heuristics 装饰器,用于指定元参数值的计算方法。
  • triton.heuristics 在自动调优不适用时非常有用。
  • 装饰器的 values 参数是一个字典,包含元参数名称和计算元参数值的函数。
➡️

继续阅读