跳到正文
原文
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