Linear Layouts: Robust Code Generation of Efficient Tensor Computation Using ${\mathbb{F}}_2$

Keren Zhou, Mario Lezcano, Adam Goucher +8

Efficient tensor computation is a cornerstone of modern deep learning (DL) workloads, yet existing approaches struggle to achieve flexible and performant design and implementation…