The tensor algebra compiler

@article{Kjolstad2017TheTA,
  title={The tensor algebra compiler},
  author={Fredrik Kjolstad and Shoaib Kamil and Stephen Chou and David Lugato and Saman P. Amarasinghe},
  journal={PACMPL},
  year={2017},
  volume={1},
  pages={77:1-77:29}
}
Tensor algebra is a powerful tool with applications in machine learning, data analytics, engineering and the physical sciences. Tensors are often sparse and compound operations must frequently be computed in a single kernel for performance and to save memory. Programmers are left to write kernels for every operation of interest, with different mixes of dense and sparse tensors in different formats. The combinations are infinite, which makes it impossible to manually implement and optimize them… CONTINUE READING

Citations

Publications citing this paper.
SHOWING 1-10 OF 44 CITATIONS

TeIL: a type-safe imperative tensor intermediate language

VIEW 12 EXCERPTS
CITES BACKGROUND & METHODS
HIGHLY INFLUENCED

Tensor Algebra Compilation with Workspaces

  • 2019 IEEE/ACM International Symposium on Code Generation and Optimization (CGO)
  • 2019
VIEW 9 EXCERPTS
CITES METHODS & BACKGROUND

A Fill Estimation Algorithm for Sparse Matrices and Tensors in Blocked Formats

  • 2018 IEEE International Parallel and Distributed Processing Symposium (IPDPS)
  • 2018
VIEW 10 EXCERPTS
CITES BACKGROUND, METHODS & RESULTS
HIGHLY INFLUENCED

Expressing Sparse Matrix Computations for Productive Performance on Spatial Architectures

  • ArXiv
  • 2018
VIEW 12 EXCERPTS
CITES BACKGROUND & METHODS
HIGHLY INFLUENCED

Fill estimation for blocked sparse matrices and tensors

VIEW 3 EXCERPTS
CITES METHODS
HIGHLY INFLUENCED

References

Publications referenced by this paper.
SHOWING 1-6 OF 6 REFERENCES

TensorFlow: A System for Large-Scale Machine Learning

Martín Abadi, Paul Barham, +19 authors Xiaoqiang Zheng
  • OSDI
  • 2016
VIEW 6 EXCERPTS
HIGHLY INFLUENTIAL

The art of computer programming: sorting and searching

Donald Ervin Knuth
  • Vol. 3. Pearson Education. Joseph C Kolecki
  • 1973
VIEW 1 EXCERPT
HIGHLY INFLUENTIAL