Verbose setting for RewritePass for better logging and debugging

Abierto
  • ¿Algo no está bien?
#2,876 2 comentarios 0 reacciones 0 asignados Ver en GitHub

Los mantenedores suelen responder en 3 días

Nadie ha tomado este issue todavía.

Evaluación

Dificultad
3/5
Tiempo estimado
1-2 días
Aptitud para principiantes
55/100
Tipo de issue
Nueva funcionalidad
Claridad
Bastante claro
Estado de actividad
Tranquilo
Stack tecnológico
python
Área
compilers

Línea de trabajo

Revisa RewritePass en la implementación del pass ONNX IR correspondiente y optimize_onnx_using_onnx_ir en tetracode/src/www/python/tetrai/compilers/onnx2onnx/optimizer.py. Sigue cómo se construye el pass y cómo se aplican las reglas; después, verifica que una configuración opcional de verbose llegue a la aplicación de reglas, mientras el comportamiento predeterminado permanece sin cambios. Se considera terminado cuando ambos puntos de entrada solicitados admiten la configuración y es observable el registro de los passes activados.

Escrito por el modelo de indexación a partir del texto del issue.

Descripción

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
          ...
      ]
Lenguaje dominante
Python
Estrellas
457
Forks
149
Merge medio
3 d 19 h
PR fusionados (30 d)
23

Preparar el entorno

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Más de microsoft/onnxscript

Todos los issues de microsoft/onnxscript

Issues similares

Más issues de Python

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.