PLDI 2023 (series) / CTSTA 2023 (series) / Compiler Techniques for Sparse Tensor Algebra / Moving the MLIR Sparse Compilation Pipeline into Production
Moving the MLIR Sparse Compilation Pipeline into ProductionVirtual
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 JunDisplayed time zone: Eastern Time (US & Canada) change
Sun 18 Jun
Displayed time zone: Eastern Time (US & Canada) change
14:00 - 15:30 | |||
14:00 15mTalk | Learning workload-aware cost model for sparse tensor program CTSTA Jaeyeon Won Massachusetts Institute of Technology | ||
14:15 15mTalk | Autoscheduling for Sparse Tensor Contraction CTSTA Kirshanthan Sundararajah Purdue University | ||
14:30 10mPanel | Discussion CTSTA | ||
14:40 15mTalk | Fantastic Sparse Masks and Where to Find Them CTSTA Shiwei Liu The University of Texas at Austin, Texas, USA | ||
14:55 15mTalk | Moving the MLIR Sparse Compilation Pipeline into ProductionVirtual CTSTA | ||
15:10 15mPanel | Discussion CTSTA | ||
15:25 5mDay closing | Closing CTSTA |