124 citations · 159 across the 5 of their papers we have counts for
5 papers
Dynamic Sparse Tensor Algebra Compilation
Stephen Chou, Saman Amarasinghe
This paper shows how to generate efficient tensor algebra code that compute on dynamic sparse tensors, which have sparsity structures that evolve over time. We propose a language f…
An Attempt to Generate Code for Symmetric Tensor Computations
Jessica Shi, Stephen Chou, Fredrik Kjolstad +1
This document describes an attempt to develop a compiler-based approach for computations with symmetric tensors. Given a computation and the symmetries of its input tensors, we der…
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…
Automatic Generation of Efficient Sparse Tensor Format Conversion Routines
Stephen Chou, Fredrik Kjolstad, Saman Amarasinghe
This paper shows how to generate code that efficiently converts sparse tensors between disparate storage formats (data layouts) such as CSR, DIA, ELL, and many others. We decompose…
Format Abstraction for Sparse Tensor Algebra Compilers
Stephen Chou, Fredrik Kjolstad, Saman Amarasinghe
This paper shows how to build a sparse tensor algebra compiler that is agnostic to tensor formats (data layouts). We develop an interface that describes formats in terms of their c…