arXiv cs.AI· David Pissarra, Jinkun Lin, Haitian Jiang, Aurojit Panda, Jinyang Li·· 9 小时前AI 评分29
Cleave:通过解耦代数搜索与算子调度扩展张量程序优化
Cleave: Scaling Tensor Program Optimization via Decoupled Algebraic Search and Operator Scheduling
AI 导读
机器学习编译器 Cleave 通过符号化解耦,将代数变换搜索与算子调度分离,在常见 LLM 子图上生成的 kernel 比最优基线最高快 2.8 倍(平均 1.6 倍),编译时间比 Mirage 平均减少 5.9 倍。
来源:arXiv cs.AI · arxiv.org