Sun 18 Jun 2023 14:55 - 15:10 at Magnolia 4 - CTSTA: Session 3

In previous work, we proposed integrating the idea of treating sparsity as a property of tensors, and not a tedious implementation task, into the MLIR compiler infrastructure. This has resulted in an open-source pipeline for sparse compilation by means of a new sparse tensor dialect. In this talk, we will discuss some interesting engineering problems we encountered while integrating the pipeline into a production compiler, in particular for the JAX framework extended with sparse tensor support.

Sun 18 Jun

Displayed time zone: Eastern Time (US & Canada) change

14:00 - 15:30
14:00
15m
Talk
Learning workload-aware cost model for sparse tensor program
CTSTA
Jaeyeon Won Massachusetts Institute of Technology
14:15
15m
Talk
Autoscheduling for Sparse Tensor Contraction
CTSTA
Kirshanthan Sundararajah Purdue University
14:30
10m
Panel
Discussion
CTSTA

14:40
15m
Talk
Fantastic Sparse Masks and Where to Find Them
CTSTA
Shiwei Liu The University of Texas at Austin, Texas, USA
14:55
15m
Talk
Moving the MLIR Sparse Compilation Pipeline into ProductionVirtual
CTSTA
Aart Bik Google, Inc., Peiming Liu Google Inc
15:10
15m
Panel
Discussion
CTSTA

15:25
5m
Day closing
Closing
CTSTA
Fredrik Kjolstad Stanford University, Saman Amarasinghe Massachusetts Institute of Technology