PyTorch Blog· Aditya Srikanth (NVIDIA), Rupak Roy (NVIDIA), Abhinav Singh (NVIDIA), Jason Ansel (Meta), Clement Anthonioz-Blanc (Meta) and Jongsok Choi (Meta)·· 12 小时前精选AI 评分63
PyTorch Helion 联合 NVIDIA CompileIQ 搜索编译器控制参数
One Search, Two Layers: Compiler Controls Inside PyTorch Helion using NVIDIA CompileIQ
AI 导读
PyTorch Helion 现在可以通过 autotune_search_acf 把 NVIDIA CompileIQ 生成的 ACF 编译器控制文件与 tile 大小、warp 数等内核配置放在同一次搜索中联合调优。
推荐理由
原文给出把 ptxas 编译控制纳入 Helion 自动调优的具体做法与实测数据，可迁移到已有内核的最后一公里优化。
来源：PyTorch Blog · pytorch.org