Verbose setting for RewritePass for better logging and debugging

オープン
  • 何か問題がありますか？
#2,876 コメント 2 件 リアクション 0 件 担当者 0 名 GitHub で見る

メンテナーはふだん 4 日以内に返信

まだ誰も着手していません。

評価

難易度
3/5
見積もり時間
1〜2日
初心者へのやさしさ
55/100
issue の種類
機能追加
明瞭さ
おおむね明確
活発さ
静か
技術スタック
python
領域
compilers

調査の方向性

関連する ONNX IR パス実装の RewritePass と、tetracode/src/www/python/tetrai/compilers/onnx2onnx/optimizer.py の optimize_onnx_using_onnx_ir を確認する。パスがどのように構築され、ルールがどのように適用されるかを追跡し、そのうえで、オプションの verbose 設定がルール適用まで渡され、デフォルトの動作が変わらないことを検証する。両方の要求されたエントリポイントがその設定をサポートし、トリガーされたパスのログを確認できれば完了とする。

索引モデルが issue の本文から書いたものです。

説明

Problem: I want a setting to know if a pass is triggered.
Solution: Make a flag in RewritePass to be verbose or not.

Here is a proposal. Is there a batter way?

Change RewritePass:
  class RewritePass(ir.passes.InPlacePass):
      def __init__(
          self,
          rules: Sequence[pattern.RewriteRule] | pattern.RewriteRuleSet,
          /,
          verbose: int = 0,  # ADD THIS
      ) -> None:
          super().__init__()
          if isinstance(rules, Sequence):
              if not rules:
                  raise ValueError("rules must not be empty")
              rules = pattern.RewriteRuleSet(rules)
          assert isinstance(rules, pattern.RewriteRuleSet)
          self.rules: pattern.RewriteRuleSet = rules
          self.verbose = verbose  # ADD THIS

      def call(self, model: ir.Model) -> ir.passes.PassResult:
          count = self.rules.apply_to_model(model, verbose=self.verbose if self.verbose else None)  # CHANGE THIS LINE
          if count:
              print(f"Applied {count} of general pattern rewrite rules.")
          return ir.passes.PassResult(model, bool(count))

  File 2: tetracode/src/www/python/tetrai/compilers/onnx2onnx/optimizer.py

  Add verbose parameter and pass it:
  def optimize_onnx_using_onnx_ir(
      onnx_model: rewriter._ModelProtoOrIr,
      rewrite_rules: Sequence[rewriter.pattern.RewriteRule] | None = None,
      logger: TetraLoggerAdapter = TetraLoggerAdapter(__name__),
      verbose: int = 0,  # ADD THIS
  ) -> rewriter._ModelProtoOrIr:
      ...
      passes = [
          ...
          rewriter.RewritePass(rewrite_rules, verbose=verbose),  # ADD verbose HERE
          ...
      ]
主要言語
Python
スター
457
フォーク
149
平均マージ
3日 18時間
マージ済み PR（30日）
23

環境構築

はじめの一歩

  1. issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
  2. 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
  3. リポジトリをフォークし、ブランチを切って変更します。
  4. issue 番号を参照したプルリクエストを送ります。

microsoft/onnxscript のほかの issue

microsoft/onnxscript の issue をすべて見る

似ている issue

Python の issue をもっと見る

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。