Sparse Tensor Optimization

Logo

Galley: Modern Query Optimization for Sparse Tensor Programs

Read the Paper
 
Explore the Code
 

About The Project

Sparse tensor programming is an emerging paradigm that attempts to generalize the highly productive tensor programming framework to a wider array of workloads. While this provides new benefits for users, it also opens up new challenges in program optimization. Sparse programs are significantly more challenging to optimize than dense programs, and their optimal form depends on the input data. Fortunately, the database community has been struggling with very similar problems for decades and has built a robust set of techniques and intuition. This project aims to bring that expertise to sparse tensor programming by building a program optimizer, named Galley, on top of the Finch compiler.

Acknowledgments

This work was supported by NSF SHF 2312195 and NSF IIS 2314527.