SPLASH 2025

In Singapore on Oct 12, 2025

Tutorial: End-to-End Compiler Infrastructure for Emerging Tensor Accelerators

Host Conference: SPLASH 2025 (co-located with ICFP 2025)
Date: Oct 12, 2025 (Sunday)
Time: 2:00 PM - 5:30 PM (UTC+8)
Venue: Seminar Room 6 (COM1-02-03), NUS School of Computing
Location: COM1 Level 2, 13 Computing Dr, Singapore - 117417
Prerequisites: Please bring your own laptop with a working installation of Docker.


Introduction

Recent years have seen a proliferation of specialized ML accelerators – proposed in both academia (e.g., Gemmini, FEATHER) and industry (e.g., Google TPU, Intel AMX) – that depart significantly from traditional CPU/GPU architectures. However, research on compiler and systems support for these accelerators remains sparse, largely due to the lack of mature open-source compiler infrastructures capable of targeting them from popular ML frameworks like PyTorch, and JAX. Building such support involves considerable manual effort, slowing innovation and creating a gap between hardware and software research communities.

This tutorial introduces the ACT (Accelerator Compiler Toolkit), an ecosystem that automatically generates complete compiler backends and essential software tooling from high-level ISA specifications of tensor accelerators.

The ACT ecosystem consists of:

  1. TAIDL (Tensor Accelerator ISA Definition Language): A Python-based DSL for specifying tensor accelerator ISAs
  2. TAIDL-TO (Test Oracle) Generator: Automatically generates fast & scalable functional simulators from TAIDL specifications
  3. ACT Backend Generator: Automatically generates sound & complete compiler backends just from TAIDL specifications
  4. XLA Integration: Enables end-to-end compilation from popular ML frameworks like JAX, TensorFlow, and PyTorch

The tutorial is designed for researchers, practitioners, and students interested in compiler design, programming languages, and AI/ML hardware. By the end, participants will have hands-on experience with the complete ACT workflow and understand how to rapidly prototype compiler support for novel accelerator architectures.


Contents and Timeline (tentative)

Time Topic Presenter
2:00 - 2:10 PM Welcome and Introduction Prof. Charith Mendis (UIUC)
2:10 - 2:30 PM Talk: Overview of ACT Ecosystem Prof. Charith Mendis (UIUC)
2:30 - 2:40 PM Tutorial Logistics Devansh Jain (UIUC)
2:40 - 2:50 PM Demonstration 1: Quick walkthrough of ACT Ecosystem Devansh Jain (UIUC)
2:50 - 3:10 PM Hands-on Exercise 1: Specifying a new Accelerator ISA Devansh Jain (UIUC)
3:10 - 3:30 PM Hands-on Exercise 2: Writing custom Accelerator Kernels Devansh Jain (UIUC)
3:30 - 4:00 PM Break  
4:00 - 4:30 PM Talk: Automatically Generating Compiler Backends just from ISA Specifications Akash Pardeshi (UIUC)
4:30 - 5:00 PM Hands-on Exercise 3: Generating a Compiler Backend for a new Accelerator ISA Devansh Jain (UIUC)
5:00 - 5:10 PM Demonstration 2: Integrating the new Accelerator Backend with XLA Compiler Devansh Jain (UIUC)
5:10 - 5:30 PM Q&A and Closing Remarks Devansh Jain (UIUC)

Organizers

  • Prof. Charith Mendis is an Assistant Professor in the Siebel School of Computing and Data Science at the University of Illinois Urbana-Champaign (UIUC). His broad research interests are at the intersection of compilers, program optimization and machine learning. He received his Ph.D. and Master’s from the Massachusetts Institute of Technology and his B.Sc. from the University of Moratuwa. He is the recipient of a DARPA Young Faculty Award, an NSF CAREER Award, the William A. Martin outstanding master’s thesis award at MIT and the university gold medal for his B.Sc. He has won numerous paper awards including a Distinguished Paper Award at POPL, a Best Student Paper Award at the IEEE BigData conference, an honorable mention for the Best Artifact Award at SIGMOD, a Best Paper Award at ML for Systems workshop at ISCA and an IEEE Top Picks Honorable Mention.
  • Devansh Jain is a Ph.D. student in the Siebel School of Computing and Data Science at the University of Illinois Urbana-Champaign (UIUC), advised by Prof. Charith Mendis. His research interests lie in the field of programming languages, compilers & computer architecture, primarily domain-specific languages and architectures. His primary research objective is to develop a unified compiler infrastructure for architectures designed for accelerating tensor computations. He has authored/co-authored multiple papers at top-tier PL & systems venues, including POPL, OOPSLA, MICRO, with a Distinguished Paper Award at POPL’25.
  • Akash Pardeshi is a M.S. student in the Department of Electrical and Computer Engineering at the University of Illinois Urbana-Champaign (UIUC), advised by Prof. Charith Mendis. His broad research interests are in compilers and computer architecture. His research focuses on techniques such as equality saturation and e-graph applications to ML compilers.