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
Discussion
Read the public discussion and references gathered around this paper.
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…