diff --git a/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorBase.td b/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorBase.td --- a/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorBase.td +++ b/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorBase.td @@ -31,9 +31,9 @@ to tensor algebra by [Kjolstad17,Kjolstad20] in the Sparse Tensor Algebra Compiler (TACO) project (see http://tensor-compiler.org). - The MLIR implementation closely follows the "sparse iteration theory" - that forms the foundation of TACO. A rewriting rule is applied to each - tensor expression in the Linalg dialect (MLIR's tensor index notation) + The MLIR implementation [Biketal22] closely follows the "sparse iteration + theory" that forms the foundation of TACO. A rewriting rule is applied to + each tensor expression in the Linalg dialect (MLIR's tensor index notation) where the sparsity of tensors is indicated using the per-dimension level types dense/compressed together with a specification of the order on the dimensions (see [Chou18] for an in-depth discussions and possible @@ -56,6 +56,10 @@ * [Bik96] Aart J.C. Bik. Compiler Support for Sparse Matrix Computations. PhD thesis, Leiden University, May 1996. + * [Biketal22] Aart J.C. Bik, Penporn Koanantakool, Tatiana Shpeisman, + Nicolas Vasilache, Bixia Zheng, and Fredrik Kjolstad. Compiler Support + for Sparse Tensor Computations in MLIR. ACM Transactions on Architecture + and Code Optimization, June, 2022 (see https://dl.acm.org/doi/10.1145/3544559). * [Chou18] Stephen Chou, Fredrik Berg Kjolstad, and Saman Amarasinghe. Format Abstraction for Sparse Tensor Algebra Compilers. Proceedings of the ACM on Programming Languages, October 2018.