跳到正文
AGI Hunt· PyTorch·· 1 小时前AI 评分62

PyTorch Helion 引入编译期启发式配置,性能反超手写 Triton 内核 18%

PyTorch Helion 编译期启发式免调优,反超手写 Triton 内核 18%

AI 导读

PyTorch 团队为 Helion 引入编译期启发式配置选择,根据内核 IR 与数据流将内核分类,并直接确定块大小、warps、stages 等参数,从而免除 GPU autotuning。

来源:AGI Hunt · agihunt.info