PyTorch· Aditya Srikanth (NVIDIA), Rupak Roy (NVIDIA), Abhinav Singh (NVIDIA), Jason Ansel (Meta), Clement Anthonioz-Blanc (Meta) and Jongsok Choi (Meta)·· 17 小时前AI 评分48
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) 实现寄存器分配与指令调度等底层优化,并支持将 ACF 列表纳入自动调优空间进行联合搜索。
来源:PyTorch · pytorch.org