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