Preprint
Jul 2026
Triton for MTIA: Bridging the Programming Model Gaps for Custom AI Accelerators
This work presents the first production-scale application of Triton on a custom ML accelerator, MTIA-2i, developed by Meta, and develops a new compiler backend that targets it, introduces enhancements to TorchInductor code generation, and proposes minimal language extensions that expose MTIA-specific architectural feat...
Haishan Zhu, Domi Yan, Michael Levesque-Dion et al.
· 0 citations