arXiv 2412.13398
Pattern Matching in AI Compilers and its Formalization (Extended Version)
By Joseph W. Cutler, Alex Collins, et al.
Published 2024-12-18
Wiki summary
Explore the paper's summary, context, and related research on Papiers.
PyPM is a Python-based domain specific language (DSL) for building rewrite-based optimization passes on machine learning computation graphs. Users define individual optimizations by writing (a) patterns that match subgraphs of a computation graph and (b) corresponding rules which replace a matched subgraph with an optimized kernel. PyPM is distinguished from the many other DSLs for defining rewriting passes by its c…