PyTorch Blog· Aditya Srikanth (NVIDIA), Rupak Roy (NVIDIA), Abhinav Singh (NVIDIA), Jason Ansel (Meta), Clement Anthonioz-Blanc (Meta) and Jongsok Choi (Meta)·· 2 小时前AI 评分47
PyTorch Helion 借助 NVIDIA CompileIQ 联合搜索编译器控制与内核配置
One Search, Two Layers: Compiler Controls Inside PyTorch Helion using NVIDIA CompileIQ
阅读原文
本站未展示全文，请前往来源网站阅读。
AI 导读
PyTorch Helion 现已接入 NVIDIA CompileIQ，可在内核自动调优之外联合搜索 ptxas 编译器控制空间，进一步优化已调优内核。CompileIQ 将结果打包为可移植的 Advanced Control File（ACF），Helion 的 autotuner 可把 ACF 列表当作与 tile 大小、warp 数并列的搜索维度。
来源：PyTorch Blog · pytorch.org