3 papers
cs.MS2021
An Asymptotic Cost Model for Autoscheduling Sparse Tensor Programs
Willow Ahrens, Fredrik Kjolstad, Saman Amarasinghe
While loop reordering and fusion can make big impacts on the constant-factor performance of dense tensor programs, the effects on sparse tensor programs are asymptotic, often leadi…
cs.DS2020
Sparse Tensor Transpositions
Suzanne Mueller, Willow Ahrens, Stephen Chou +2
We present a new algorithm for transposing sparse tensors called Quesadilla. The algorithm converts the sparse tensor data structure to a list of coordinates and sorts it with a fa…
cs.MS2018
Sparse Tensor Algebra Optimizations with Workspaces
Fredrik Kjolstad, Willow Ahrens, Shoaib Kamil +1
This paper shows how to optimize sparse tensor algebraic expressions by introducing temporary tensors, called workspaces, into the resulting loop nests. We develop a new intermedia…