PyTorch Blog· Aditya Srikanth (NVIDIA), Rupak Roy (NVIDIA), Abhinav Singh (NVIDIA), Jason Ansel (Meta), Clement Anthonioz-Blanc (Meta) and Jongsok Choi (Meta)·· 1 天前精选AI 评分60
PyTorch Helion 利用 NVIDIA CompileIQ 联合搜索内核与编译器控制项
One Search, Two Layers: Compiler Controls Inside PyTorch Helion using NVIDIA CompileIQ
AI 导读
PyTorch Helion 现在可通过 NVIDIA CompileIQ，将内核配置与 PTXAS 的 Advanced Control File（ACF）编译器控制项放在同一轮 autotuning 中搜索。
推荐理由
文章把 Helion 的内核配置搜索与 PTXAS 编译器控制联系起来，并用可复现实验展示 ACF 在已调优 kernel 上继续挖掘性能的路径。
来源：PyTorch Blog · pytorch.org