Fri 17 Jun 2022 16:30 - 16:50 at Cockatoo - Rewriting Chair(s): Chandrakana Nandi

We present a lightweight Coq framework for optimizing tensor kernels written in a pure, functional array language. Optimizations rely on user scheduling using series of verified, semantics-preserving rewrites. Unusually for compilation targeting imperative code with arrays and nested loops, all rewrites are source-to-source within a purely functional language. Our language comprises a set of core constructs for expressing high-level computation detail and a set of what we call reshape operators, which can be derived from core constructs but trigger low-level decisions about storage patterns and ordering. We demonstrate that not only is this system capable of deriving the optimizations of existing state-of-the-art languages like Halide and generating comparably performant code, it is also able to schedule a family of useful program transformations beyond what is reachable in Halide.

Fri 17 Jun

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

15:30 - 16:50
RewritingSIGPLAN Track at Cockatoo
Chair(s): Chandrakana Nandi Certora, inc.
15:30
20m
Talk
(ICFP 2020) Achieving high-performance the functional way: a functional pearl on expressing high-performance optimizations as rewrite strategies
SIGPLAN Track
Bastian Hagedorn NVIDIA, Johannes Lenfers University of Münster, Thomas Koehler University of Glasgow, Xueying Qin University of Glasgow, United Kingdom, Sergei Gorlatch University of Münster, Germany, Michel Steuwer University of Edinburgh
Link to publication DOI Authorizer link Pre-print
15:50
20m
Talk
(PLDI 2020) Optimizing Homomorphic Evaluation Circuits by Program Synthesis and Term Rewriting
SIGPLAN Track
DongKwon Lee Seoul National University, South Korea, Woosuk Lee Hanyang University, Hakjoo Oh Korea University, Kwangkeun Yi Seoul National University
16:10
20m
Talk
(ICFP 2021) Grafs: Declarative Graph Analytics
SIGPLAN Track
Farzin Houshmand University of California, Riverside, Mohsen Lesani University of California at Riverside, Keval Vora Simon Fraser University
16:30
20m
Talk
(POPL 2022) Verified Tensor-Program Optimization Via High-level Scheduling Rewrites
SIGPLAN Track
Amanda Liu Massachusetts Institute of Technology, Gilbert Bernstein University of California at Berkeley, Adam Chlipala MIT CSAIL, Jonathan Ragan-Kelley MIT