Compiler Support for Sparse Tensor Computations in MLIR

Aart Bik; Penporn Koanantakool; Tatiana Shpeisman; Nicolas Vasilache; Bixia Zheng; Fredrik Kjolstad

doi:10.1145/3544559

ACM TACO · September 2022

Compiler Support for Sparse Tensor Computations in MLIR

Aart BikPenporn KoanantakoolTatiana ShpeismanNicolas VasilacheBixia Zheng Fredrik Kjolstad

ACM Transactions on Architecture and Code Optimization

Read PDF DOI Video

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

@article{bik2022compiler,
  author    = {Bik, Aart and Koanantakool, Penporn and Shpeisman, Tatiana and Vasilache, Nicolas and Zheng, Bixia and Kjolstad, Fredrik},
  title     = {Compiler Support for Sparse Tensor Computations in {MLIR}},
  journal   = {ACM Transactions on Architecture and Code Optimization},
  year      = {2022},
  month     = sep,
  volume    = {19},
  number    = {4},
  articleno = {50},
  numpages  = {25},
  publisher = {Association for Computing Machinery},
  address   = {New York, NY, USA},
  doi       = {10.1145/3544559},
  url       = {https://doi.org/10.1145/3544559}
}

Citation

Aart Bik, Penporn Koanantakool, Tatiana Shpeisman, Nicolas Vasilache, Bixia Zheng, and Fredrik Kjolstad. Compiler Support for Sparse Tensor Computations in MLIR. ACM TACO, 2022.

All publications