An Attempt to Generate Code for Symmetric Tensor Computations
Abstract
This document describes an attempt to develop a compilerbased approach for computations with symmetric tensors. Given a computation and the symmetries of its input tensors, we derive formulas for random access under a storage scheme that eliminates redundancies; construct intermediate representations to describe the loop structure; and translate this information, using the taco tensor algebra compiler, into code. While we achieve a framework for reasoning about a fairly general class of symmetric computations, the resulting code is not performant when the symmetries are misaligned.
 Publication:

arXiv eprints
 Pub Date:
 September 2021
 DOI:
 10.48550/arXiv.2110.00186
 arXiv:
 arXiv:2110.00186
 Bibcode:
 2021arXiv211000186S
 Keywords:

 Computer Science  Mathematical Software;
 Computer Science  Programming Languages