跳到正文
arXiv cs.AI· David Pissarra, Jinkun Lin, Haitian Jiang, Aurojit Panda, Jinyang Li·· 5 小时前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