[論文レビュー] egg: Easy, Efficient, and Extensible E-graphs.

Max Willsey, Yisu Remy Wang+4 more|arXiv (Cornell University)|Apr 7, 2020

Software Testing and Debugging Techniques被引用数 3

ひとこと要約

この論文では、性能向上のための再構築による償済的不変条件の回復と、拡張性のためのe-class解析を特徴とする、オープンソースのe-グラフライブラリeggを紹介する。その結果、プログラム合成およびコンパイラ最適化の分野で最先端の結果を達成する、より高速で柔軟なe-グラフが実現された。

ABSTRACT

An e-graph efficiently represents a congruence relation over many expressions. Although they were originally developed in the late 1970s for use in automated theorem provers, a more recent technique known as equality saturation repurposes e-graphs to implement state-of-the-art, rewrite-driven compiler optimizations and program synthesizers. However, e-graphs remain unspecialized for this newer use case. Equality saturation workloads exhibit distinct characteristics and often require ad-hoc e-graph extensions to incorporate transformations beyond purely syntactic rewrites. This work contributes two techniques that make e-graphs fast and extensible, specializing them to equality saturation. A new amortized invariant restoration technique called rebuilding takes advantage of equality saturation's distinct workload, providing asymptotic speedups over current techniques in practice. A general mechanism called e-class analyses integrates domain-specific analyses into the e-graph, reducing the need for ad hoc manipulation. We implemented these techniques in a new open-source library called egg. Our case studies on three previously published applications of equality saturation highlight how egg's performance and flexibility enable state-of-the-art results across diverse domains.

研究の動機と目的

• 等価性飽和ワークロードに適用された際の、既存のe-グラフの性能および拡張性の制限を解決すること。
• 現在のe-グラフが活用していない、等価性飽和ワークロードの特徴を同定すること。
• 等価性飽和ワークロードのワークロードパターンを活用して、不変条件の回復をより効率的に行う技術を開発すること。
• コアデータ構造へのアドホックな修正を必要とせず、ドメイン固有の解析をe-グラフに統合する一般化されたメカニズムを構築すること。
• 提案手法が、プログラム合成およびコンパイル分野における多様な応用で最先端の結果を達成できることを実証すること。

提案手法

• 等価性飽和ワークロードの繰り返し的かつ構造的な性質を活用し、複数の操作にわたるコストを償済することで、漸近的な高速化を達成する再構築を導入する。
• ドメイン固有の知識をe-グラフノードに直接埋め込む一般化されたメカニズムとしてe-class解析を設計する。これにより、低レベルのアドホックな操作への依存を低減する。
• 等価性飽和下でのe-グラフの効率的構築と照会をサポートする、オープンソースのe-グラフライブラリeggとして技術を実装する。
• 特に同値性および標準形照会に対して、インクリメンタルな更新と効率的な照会解決を可能にするようにe-グラフ操作を構造化する。
• e-class解析をe-グラフの構築および変換中に組み合わせて実行できる、一等身のコンponentsとして統合する。
• 高スループットでリライトに依存するワークロードにおけるオーバーヘッドを最小限に抑えるために、eggのデータ構造とアルゴリズムを最適化する。

実験結果

リサーチクエスチョン

• RQ1 等価性飽和ワークロードの構造的パターンを活用して、償済的不変条件回復技術を設計することで、性能を向上させることができるか？
• RQ2 コアデータ構造へのアドホックな修正を必要とせず、e-グラフにドメイン固有の解析を体系的に統合する方法は何か？
• RQ3 提案手法が、実世界の等価性飽和応用において、性能および柔軟性をどの程度向上させられるか？
• RQ4 プログラム合成およびコンパイラ最適化における多様なユースケースをサポートできる統一的かつ拡張可能なe-グラフライブラリを構築できるか？

主な発見

• 再構築技術は、複数の操作にわたるコストを償済することで、従来の不変条件回復手法よりも漸近的な高速化を実現する。
• e-class解析により、ドメイン固有の論理をe-グラフ構造に直接統合できるため、アドホックなe-グラフ操作の必要性が低下する。
• eggライブラリは、以前に発表された3つの等価性飽和応用において、最先端の性能と柔軟性を達成している。
• 再構築とe-class解析の組み合わせにより、非文法的変換を効率的に処理できるようになった。
• 事例研究により、eggが大規模なプログラム合成およびコンパイラ最適化を高パフォーマンスでサポートできることを示した。
• eggのオープンソース性により、e-グラフを必要とする多様な分野における再利用と拡張が可能になった。