PyTorch Blog· Aditya Srikanth (NVIDIA), Rupak Roy (NVIDIA), Abhinav Singh (NVIDIA), Jason Ansel (Meta), Clement Anthonioz-Blanc (Meta) and Jongsok Choi (Meta)·原文发布· 4 小时前AI 评分46
One Search, Two Layers: 用 NVIDIA CompileIQ 在 PyTorch Helion 中控制编译器
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 数等一起联合搜索。
本站提供 AI 导读，完整报道请前往来源网站阅读。
阅读原文 PyTorch Blog · pytorch.org