An Attempt to Generate Code for Symmetric Tensor Computations
Abstract
This document describes an attempt to develop a compiler-based 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 e-prints
- Pub Date:
- September 2021
- DOI:
- arXiv:
- arXiv:2110.00186
- Bibcode:
- 2021arXiv211000186S
- Keywords:
-
- Computer Science - Mathematical Software;
- Computer Science - Programming Languages