arXiv · Machine Learning· Bohan Hou, Hongyi Jin, Guanjie Wang, Shushi Hong, Jinqi Chen, Yaxing Cai, Lijie Yang, Zihao Ye, Yaoyao Ding, Ruihang Lai, Todd Mowry, Tianqi Chen·· 8 小时前AI 评分46
TIRx：面向异构后端的统一 Tile Primitive 机器学习编译器
TIRx: A Unified Tile Primitive ML Compiler Across Heterogeneous Backends
AI 导读
TIRx 是一款面向 GPU 和 AI 加速器的机器学习编译器，通过显式接口统一处理不同硬件的执行范围、张量布局与 Tile Primitive 分发。它根据执行范围、布局和后端将高级操作 lowering 为高效实现，同时保留底层控制能力。评估显示，其在新一代 GPU 和 AI 加速器上达到厂商级性能，并显著减少底层实现代码。
来源：arXiv · Machine Learning · arxiv.org