Tensor Algebra Compilation with Workspaces

Fredrik Kjolstad; Willow Ahrens; Shoaib Kamil; Saman Amarasinghe

doi:10.1109/CGO.2019.8661185

CGO · February 2019

Tensor Algebra Compilation with Workspaces

Fredrik Kjolstad Willow Ahrens Shoaib Kamil Saman Amarasinghe

IEEE/ACM International Symposium on Code Generation and Optimization

Read PDF DOI

Topics Sparse tensor algebra Compiler representations

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

@inproceedings{kjolstad2019tensor,
  author    = {Kjolstad, Fredrik and Ahrens, Willow and Kamil, Shoaib and Amarasinghe, Saman},
  title     = {Tensor Algebra Compilation with Workspaces},
  booktitle = {2019 {IEEE}/{ACM} International Symposium on Code Generation and Optimization ({CGO})},
  year      = {2019},
  month     = feb,
  pages     = {180--192},
  publisher = {IEEE},
  doi       = {10.1109/CGO.2019.8661185},
  url       = {https://doi.org/10.1109/CGO.2019.8661185}
}

Citation

Fredrik Kjolstad, Willow Ahrens, Shoaib Kamil, and Saman Amarasinghe. Tensor Algebra Compilation with Workspaces. CGO, 2019.

All publications