Verbose setting for RewritePass for better logging and debugging

未关闭
  • 哪里不对？
#2,876 2 条评论 0 个 reaction 已指派 0 人 在 GitHub 查看

维护者通常 4 天内回复

还没有人认领这个 Issue。

评估

难度
3/5
预计耗时
1-2 天
新手友好度
55/100
Issue 类型
功能
描述清晰度
基本清楚
活跃度
冷清
技术栈
python
领域
compilers

调研方向

检查相关 ONNX IR pass 实现中的 RewritePass，以及 tetracode/src/www/python/tetrai/compilers/onnx2onnx/optimizer.py 中的 optimize_onnx_using_onnx_ir。跟踪 pass 的构造方式和规则的应用方式，然后验证可选的 verbose 设置能够传递到规则应用，同时默认行为保持不变。当两个要求的入口点都支持该设置，并且可以观察到触发的 pass 日志时，即表示完成。

由索引模型根据 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 天 16 小时
30 天内合并 PR
24

环境准备

  • 没有 Dockerfile 或 Docker Compose 文件
  • 没有 Pull Request 模板
  • 阅读贡献指南

从这里开始

  1. 先读完整个 Issue，再读项目的贡献指南。
  2. 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
  3. Fork 仓库，在一个分支上完成修改。
  4. 提交 Pull Request，并在描述里引用这个 Issue 编号。

microsoft/onnxscript 的其他 Issue

查看 microsoft/onnxscript 的全部 Issue

相似的 Issue

更多 Python Issue

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。