arXiv:2604.11109cs.DCcs.AI2026-04被引 1

用进化搜索自动优化GPU内核,速度更快、效果更好。

Record-Remix-Replay: Hierarchical GPU Kernel Optimization using Evolutionary Search

论文配图:Record-Remix-Replay: Hierarchical GPU Kernel Optimization using Evolutionary Search
图 1 · 摘自论文原文
  • 分层搜索源码到编译器参数的优化空间
  • 比传统方法提升性能,速度接近快10倍
  • 适合需要高效调优的科学计算开发者

随着高性能计算和人工智能工作负载对GPU依赖加深,如何在快速迭代的硬件上保持高性能成为重大挑战。开发者往往需数月时间调优科学应用以充分挖掘新架构潜力,面对从算法设计到编译器选项、代码生成顺序及内核启动参数的复杂优化空间。现有方法虽可独立搜索部分空间(如启动配置或编译选项),但全面优化仍需大量人工经验与反复试错。本文提出Record-Remix-Replay(R^3)框架,结合大模型驱动的进化搜索、贝叶斯优化与记录-重放编译技术,实现从源码级选择到编译器流程排序及运行时配置的端到端高效探索。通过加速候选方案评估,该方法可实际覆盖通常被分开处理的优化维度。实验表明,R^3在内核参数与编译器标志上优化完整科学应用的效果优于传统方法,且相比现代进化搜索提速近一个数量级。

原文摘要 · Abstract (English)

As high-performance computing and AI workloads become increasingly dependent on GPUs, maintaining high performance across rapidly evolving hardware generations has become a major challenge. Developers often spend months tuning scientific applications to fully exploit new architectures, navigating a complex optimization space that spans algorithm design, source implementation, compiler flags and pass sequences, and kernel launch parameters. Existing approaches can effectively search parts of this space in isolation, such as launch configurations or compiler settings, but optimizing across the full space still requires substantial human expertise and iterative manual effort. In this paper, we present Record-Remix-Replay (R^3), a hierarchical optimization framework that combines LLM-driven evolutionary search, Bayesian optimization, and record-replay compilation techniques to efficiently explore GPU kernel optimizations from source-level implementation choices down to compiler pass ordering and runtime configuration. By making candidate evaluation fast and scalable, our approach enables practical end-to-end search over optimization dimensions that are typically treated separately. We show that Record-Remix-Replay can optimize full scientific applications better than traditional approaches over kernel parameters and compiler flags, while also being nearly an order of magnitude faster than modern evolutionary search approaches.

GPU优化进化搜索编译器科学计算

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。