kirancodes.me
To Proof Maintenance & Beyond!
ASE 2017

taco: a tool to generate tensor algebra kernels

Fredrik Kjolstad, Stephen Chou, David Lugato, Shoaib Kamil, Saman P. Amarasinghe

Full textDOI 10.1109/ase.2017.8115709dblp

Abstract

Tensor algebra is an important computational abstraction that is increasingly used in data analytics, machine learning, engineering, and the physical sciences. However, the number of tensor expressions is unbounded, which makes it hard to develop and optimize libraries. Furthermore, the tensors are often sparse (most components are zero), which means the code has to traverse compressed formats. To support programmers we have developed taco, a code generation tool that generates dense, sparse, and mixed kernels from tensor algebra expressions. This paper describes the taco web and command-line tools and discusses the benefits of a code generator over a traditional library. See also the demo video at tensor-compiler.org/ase2017.

BibTeX
@inproceedings{Kjolstad-al:ASE17,
  author    = {Fredrik Kjolstad and
               Stephen Chou and
               David Lugato and
               Shoaib Kamil and
               Saman P. Amarasinghe},
  title     = {taco: a tool to generate tensor algebra kernels},
  booktitle = {ASE},
  pages     = {943--948},
  publisher = {{IEEE} Computer Society},
  year      = {2017},
}

Related papers