TIRx: A Unified Tile Primitive ML Compiler Across Heterogeneous Backends
TIRx is presented, an ML compiler for GPUs and AI accelerators that exposes these ingredients through explicit interfaces and introduces tile-primitive dispatch, which lowers high-level operations to efficient implementations according to scope, layout, and backend.
Bo-Han Hou, H. Jin, Guan-Jie Wang et al.
· 6 citations
· ⚡2