Abstract:Sparse matrix operations involve a large number of zero operands which makes most of the operations redundant. The amount of redundancy magnifies when a matrix operation repeatedly executes on sparse data. Optimizing matrix operations for sparsity involves either reorganization of data or reorganization of computations, performed either at compile-time or run-time. Although compile-time techniques avert from introducing run-time overhead, their application either is limited to simple sparse matrix operations generating dense output and handling immutable sparse matrices or requires manual intervention to customize the technique to different matrix operations. We contribute a compile time technique called SpComp that optimizes a sparse matrix operation by automatically customizing its computations to the positions of non-zero values of the data. Our approach neither incurs any run-time overhead nor requires any manual intervention. It is also applicable to complex matrix operations generating sparse output and handling mutable sparse matrices. We introduce a data-flow analysis, named Essential Indices Analysis, that statically collects the symbolic information about the computations and helps the code generator to reorganize the computations. The generated code includes piecewise-regular loops, free from indirect references and amenable to further optimization. We see a substantial performance gain by SpComp-generated SpMSpV code when compared against the state-of-the-art TACO compiler and piecewise-regular code generator. On average, we achieve 79% performance gain against TACO and 83% performance gain against the piecewise-regular code generator. When compared against the CHOLMOD library, SpComp generated sparse Cholesky decomposition code showcases 65% performance gain on average.

Compiling Recurrences over Dense and Sparse Arrays

Compilation of Modular and General Sparse Workspaces

Automatic generation of efficient sparse tensor format conversion routines

SpComp: A Sparsity Structure-Specific Compilation of Matrix Operations

Automatic Compiler-Based Data Structure Generation

SparseAuto: An Auto-Scheduler for Sparse Tensor Computations Using Recursive Loop Nest Restructuring

SySTeC: A Symmetric Sparse Tensor Compiler

A General Pattern-Based Dynamic Compilation Framework for Coarse-Grained Reconfigurable Architectures.

ReACT: Redundancy-Aware Code Generation for Tensor Expressions.

SpDISTAL: Compiling Distributed Sparse Tensor Computations

Automatic Generation of Efficient Linear Algebra Programs

Code Generation for a Variety of Accelerators for a Graph DSL

Automatically harnessing sparse acceleration

Intelligent-Unrolling: Exploiting Regular Patterns in Irregular Applications

Linnea: Automatic Generation of Efficient Linear Algebra Programs

Systolic Computing on GPUs for Productive Performance

Polyhedral Specification and Code Generation of Sparse Tensor Contraction with Co-iteration

Software for Sparse Tensor Decomposition on Emerging Computing Architectures

Sgap: Towards Efficient Sparse Tensor Algebra Compilation for GPU

Sparse-HeteroCL: from Sparse Tensor Algebra to Highly Customized Accelerators on FPGAs.

A Compile/Run-time Environment for the Automatic Transformation of Linked List Data Structures