7 citations · 13 across the 15 of their papers we have counts for
4 papers · 2 filters
Tensor Seeks Layout: Formalizing Layout Selection for ML Compilers
Clemens Eisenhofer, Yuwen Jia, Daniel Kroening +1
Modern machine learning compilers select tensor memory layouts to minimize execution cost under hardware constraints. Layout selection is global: an operator may be fastest under o…
A Barrier-Free Synchronization Algorithm for Multi-Engine AI Accelerators
Chungha Sung, Nikil V. Shyamsunder, Hanliang Zhang +2
Multi-engine AI accelerators such as AWS Trainium comprise specialized compute engines that execute in parallel, and the compiler must synchronize the data dependencies between the…
Mostly Automatic Translation of Language Interpreters from C to Safe Rust
Bo Wang, Brandon Paulsen, Joey Dodds +3
Translating C programs to safe Rust is challenging owing to significant differences in typing constraints, ownership, and borrowing rules. Interpreter programs are particularly imp…
Axon: A Synthesizing Superoptimizer for Tensor Programs
Akash Kothari, Shaowei Zhu, Daniel Kroening +1
Writing high performance kernels for AI accelerators requires deep expertise in tiling, instruction selection, data layout, and operator fusion placing a significant burden on prog…