PyTorch Helion 借助 NVIDIA CompileIQ 联合搜索 ptxas 编译控制
💡 灵机 AI 深度洞见与核心提炼
PyTorch Helion 现可借助 NVIDIA CompileIQ 搜索 ptxas 编译控制空间,并将其与 tile 大小、warp 数等内核配置联合调优,在已离线调优的约 27 个内核上进一步提速。CompileIQ 把调优结果打包为可移植的 Advanced Control File(ACF),Helion 的 autotuner 可将 ACF 列表作为额外搜索维度评估。
信源媒体:PyTorch:Blog(RSS)
访问出处网页 ↗