观点 / 方法普通
PyTorch Helion 借助 NVIDIA CompileIQ 联合搜索 ptxas 编译控制
原始标题:One Search, Two Layers: Compiler Controls Inside PyTorch Helion using NVIDIA CompileIQ
内容摘要
PyTorch Helion 现可借助 NVIDIA CompileIQ 搜索 ptxas 编译控制空间,并将其与 tile 大小、warp 数等内核配置联合调优,在已离线调优的约 27 个内核上进一步提速。CompileIQ 把调优结果打包为可移植的 Advanced Control File(ACF),Helion 的 autotuner 可将 ACF 列表作为额外搜索维度评估。