GPU Kernel Optimization World Model LLM-guided Search Automated Planning
摘要

针对 GPU 内核优化中复杂设计因素交织及硬件快速演进的挑战,现有方法常因缺乏显式规划能力而难以处理需多步结构变换的复杂内核。本文提出 K-Search,基于协同进化世界模型框架,利用 LLM 领域知识引导搜索空间探索。该方法显式解耦高层算法规划与底层程序实例化,使系统能 navigating 非单调优化路径并容忍临时实现缺陷。实验表明,K-Search 在 FlashInfer 多种复杂内核上显著优于最先进演化搜索方法,平均提升 2.10 倍,最高达 14.3 倍。

AI 推荐理由

论文核心在于解耦高层算法规划与底层实例化,利用世界模型指导多步优化路径搜索。

研究机构
UC Berkeley
论文信息
作者 Shiyi Cao, Ziming Mao, Joseph E. Gonzalez, Ion Stoica
发布日期 2026-02-22
arXiv ID 2602.19128
相关性评分 9/10 (高度相关)