The Tensor Algebra Compiler

Fredrik Kjolstad; Shoaib Kamil; Stephen Chou; David Lugato; Saman Amarasinghe

doi:10.1145/3133901

OOPSLA · October 2017

The Tensor Algebra Compiler

Fredrik Kjolstad Shoaib Kamil Stephen Chou David Lugato Saman Amarasinghe

ACM SIGPLAN Conference on Object-Oriented Programming, Systems, Languages, and Applications · PACMPL vol. 1, OOPSLA

Distinguished Paper Award

Read PDF DOI Video

Topics Sparse tensor algebra

Related software

TACO: The Tensor Algebra Compiler A compiler that generates fast kernels for tensor algebra expressions over dense and sparse tensors from a high-level index notation.

Cite this paper

BibTeX

@article{kjolstad2017tensor,
  author    = {Kjolstad, Fredrik and Kamil, Shoaib and Chou, Stephen and Lugato, David and Amarasinghe, Saman},
  title     = {The Tensor Algebra Compiler},
  journal   = {Proceedings of the ACM on Programming Languages},
  year      = {2017},
  month     = oct,
  volume    = {1},
  number    = {OOPSLA},
  articleno = {77},
  numpages  = {29},
  publisher = {Association for Computing Machinery},
  address   = {New York, NY, USA},
  doi       = {10.1145/3133901},
  url       = {https://doi.org/10.1145/3133901}
}

Citation

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

All publications