arXiv · AI· Bohan Hou, Hongyi Jin, Guanjie Wang, Shushi Hong, Jinqi Chen, Yaxing Cai, Lijie Yang, Zihao Ye, Yaoyao Ding, Ruihang Lai, Todd Mowry, Tianqi Chen·· 4 小时前AI 评分29
TIRx：面向异构后端的统一 Tile 原语 ML 编译器
TIRx: A Unified Tile Primitive ML Compiler Across Heterogeneous Backends
AI 导读
TIRx 是一款面向 GPU 和 AI 加速器的 ML 编译器，通过引入显式执行域抽象、张量布局映射及 Tile 原语分发机制，在保留原生性能控制的同时自动化了底层代码生成。 其设计通过命名轴和分片/副本/偏移语义将逻辑张量映射至异构硬件空间，根据执行域、布局和后端选择高效实现。 评估结果显示，TIRx 在最新 GPU 和 AI 加速器上达到了厂商级性能，并显著减少了底层实现代码量。
来源：arXiv · AI · arxiv.org