Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces a novel generative method for provenance tracking in AI compilers that leverages observational semantics rather than traditional identifier propagation. By employing a coalgebraic model and bisimulation, the authors ensure that provenance is preserved even in the presence of non-injective graph rewrites, addressing limitations of existing solutions. The implementation in the COVAN prototype demonstrates reliable provenance tracking across compilation pipelines with minimal engineering effort, significantly enhancing the debugging and validation processes in AI compilation.
Tracking tensor provenance in AI compilers just got a whole lot easier—without invasive methods or heavy overhead.
AI compilers aggressively rewrite computation graphs through normalization, lowering, and optimization, making it difficult to track the provenance of tensors and operators across compilation. Reliable provenance is essential for attaching platform-specific postprocessing, debugging compiler behavior, and validating transformations, yet existing solutions are either invasive or ad hoc under non-injective graph rewrites. We present a lightweight, generative approach to provenance tracking based on observational semantics. Instead of propagating identifiers through compiler passes, we observe graph transformations and reason about provenance in terms of observable computational actions. We formalize this approach using a coalgebraic model and bisimulation, which preserves provenance even when intermediate nodes are eliminated. Furthermore, we implement this approach in a prototype AI compiler COVAN, demonstrating stable provenance across compilation pipelines with minimal engineering overhead.