AI RESEARCH
PolyBlocks: A Compiler Infrastructure for AI Chips and Programming Frameworks
arXiv CS.LG
•
ArXi:2603.06731v1 Announce Type: cross We present the design and implementation of PolyBlocks, a modular and reusable MLIR-based compiler infrastructure for AI programming frameworks and AI chips. PolyBlocks is based on pass pipelines that compose transformations on loop nests and SSA, primarily relying on lightweight affine access analysis; the transformations are stitched together in specialized ways to realize high-performance code automatically by the use of analytical cost models and heuristics.