Compiler Optimization via LLM Reasoning for Efficient Model Serving
论文把模型服务中的编译器调优建模为上下文相关的序列决策问题:LLM 根据当前程序状态和历史性能反馈提出硬件感知变换,再由结构化蒙特卡洛树搜索平衡探索与利用。作者强调无需重新训练 LLM,即可在神经网络编译优化中提升样本效率,并以更少的候选评估获得与领先神经编译器相当或更好的加速效果。
INSIGHT对推理基建团队而言,LLM 更适合做离线编译搜索器,而不是进入在线请求路径。可将 kernel 融合、布局变换、并行策略等优化封装成可验证动作,用 MCTS 保留搜索多样性,再把最优配置固化进运行时;上线前必须复测不同模型、输入长度和硬件,避免把一次性的编译收益误当成稳定服务收益。