Track nodes added in pattern builder

onnxscript/rewriter/cast_constant_of_shape.py

@@ -9,7 +9,6 @@
 from onnxscript import ir
 from onnxscript.rewriter import pattern
 
-op = pattern.onnxop
 logger = logging.getLogger(__name__)
 
 

onnxscript/rewriter/gemm_to_matmul_add.py

@@ -3,8 +3,6 @@
 from onnxscript.rewriter import pattern
 from onnxscript.rewriter.broadcast_to_matmul import check_if_not_need_reshape
 
-op = pattern.onnxop
-
 
 # Pattern to match against
 def reshape_gemm_reshape_pattern(op, input_a, input_b, input_c, shape_a, shape_c):

onnxscript/rewriter/llama_rule_sets.py

@@ -11,8 +11,6 @@
 import onnxscript.rewriter.no_op as no_op
 import onnxscript.rewriter.pattern as orp
 
-op = orp.onnxop
-
 
 class CastIdentity(orp.RewriteRuleAsClass):
     """Replaces ``Cast(., to=to)`` by ``Identity`` if possible."""

onnxscript/rewriter/no_op.py

@@ -2,8 +2,6 @@
 # Licensed under the MIT License.
 from onnxscript.rewriter import pattern
 
-op = pattern.onnxop
-
 # TODO: Support 1-D constant tensors
 # https://github.com/microsoft/onnx-rewriter/issues/186
 

onnxscript/rewriter/onnxruntime/fused_matmul_rule_sets.py

@@ -6,8 +6,6 @@
 
 import onnxscript.rewriter.pattern as orp
 
-op = orp.onnxop
-
 
 class FusedMatMulDiv1(orp.RewriteRuleAsClass):
     """Replaces ``MatMul + Div`` by FusedMatMul."""

onnxscript/rewriter/onnxruntime/softmax.py

@@ -9,7 +9,6 @@
 from onnxscript import ir
 from onnxscript.rewriter import pattern
 
-op = pattern.onnxop
 logger = logging.getLogger(__name__)