跳到正文
arXiv cs.AI· Aheli Poddar, Sanskar Prasad, Arindam Samanta, Subha Chakraborty, Vishal Goyal, Rohit Singh Rathaur·· 9 小时前AI 评分29

KernelOPT:面向 GPU Kernel 优化的调度感知智能体搜索

KernelOPT: Dispatch-Aware Agentic Search for GPU Kernel Optimization

AI 导读

KernelOPT 是一个多智能体系统,将编译后的模型视为结构化产物,保留 cuBLAS、cuDNN 等厂商库调用,仅针对生成的 Triton 子 kernel,由五个 profiling 引导的 LLM 智能体进行优化。

来源:arXiv cs.AI · arxiv.org