Home
Home
  Research
    Datatypes
    Immutator
  Old Projects
  Calendar
  CV

 Publications
Fredrik Kjolstad, Shoaib Kamil, Stephen Chou, David Lugato and Saman Amarasinghe
  The Tensor Algebra Compiler

Technical Report, MIT, February 2017.

Abstract

Tensor and linear algebra is pervasive in data analytics and the physical sciences. Often the tensors, matrices or even vectors are sparse. Computing expressions involving a mix of sparse and dense tensors, matrices and vectors requires writing kernels for every operation and combination of formats of interest. The number of possibilities is infinite, which makes it impossible to write library code for all. This problem cries out for a compiler approach. This paper presents a new technique that compiles compound tensor algebra expressions combined with descriptions of tensor formats into efficient loops. The technique is evaluated in a prototype compiler called taco, demonstrating competitive performance to best-in-class hand-written codes for tensor and matrix operations.

Documents

download article:

BibTeX

@techreport{kjolstad17:tensor-compiler-techreport,
  author={Fredrik Kjolstad, Shoaib Kamil Stephen Chou, David Lugato and Saman Amarasinghe},
  title={{The Tensor Algebra Compiler}},
  institution={Massachusetts Institute of Technology},
  year={2017},
  month={February},
  publisher={CSAIL},
  source={http://hdl.handle.net/1721.1/107013}
}