arXiv 机器学习· Yihan Wang, Peiyu Liu, Runyu Chen, Wei Xu·· 6 小时前AI 评分37
AutoProfiler：面向 GPU kernel 优化的 profiling 选择学习
Less is more: learning what to profile for GPU kernel optimization
AI 导读
研究人员提出 AutoProfiler 框架，通过优化反馈构建确定性 profiling tree，为 GPU kernel 自适应选择硬件 profiling 信息。该框架借助 REWRITE 和 SPLIT 操作迭代优化，在 KernelBench 测试中于受限 profiling 预算下取得了优于固定配置的 kernel 优化表现。目前相关代码已开源。
来源：arXiv 机器学习 · arxiv.org