Canonicalization of Batched Einstein Summations for Tuning Retrieval
arXiv:2601.12220v1 Announce Type: new Abstract: We present an algorithm for normalizing emph{Batched Einstein Summation} expressions by mapping mathematically equivalent formulations to a unique normal form. Batches of einsums with the same Einstein notation that exhibit substantial data reuse appear frequently in finite element methods (FEM), numerical linear algebra, and computational chemistry. To effectively exploit this temporal locality for high performance, we consider groups of einsums in batched form. Representations of equivalent batched einsums may differ due to index […]