remove unused nodes call

Author: shubhambhokare1

onnxscript/rewriter/ort_fusions/cos_sin_cache.py

@@ -5,7 +5,6 @@
 import numpy as np
 
 import onnxscript.ir as ir
-from onnxscript.optimizer import remove_unused_nodes
 from onnxscript.rewriter import _fusion_utils, _ir_utils, pattern
 
 # Rewrite the computation of cos/sin cache into the form expected by ORT's custom ops.
@@ -169,9 +168,4 @@ def rewrite(
 cos_sin_cache_rules = pattern.RewriteRuleSet([_cast, _cast_const_freqs, _const_freqs, _basic])
 
 
-def fuse_cos_sin_cache(model: ir.Model, debug: bool = False) -> int:
-    fuse_cos_sin_cache = _fusion_utils.apply_fusion_rules(cos_sin_cache_rules)
-    count = fuse_cos_sin_cache(model, debug=debug)
-    if count != 0:
-        remove_unused_nodes(model)
-    return count
+fuse_cos_sin_cache = _fusion_utils.apply_fusion_rules(cos_sin_cache_rules)

onnxscript/rewriter/ort_fusions/gqa.py

@@ -2,8 +2,6 @@
 # Licensed under the MIT License.
 from __future__ import annotations
 
-import onnxscript.ir as ir
-from onnxscript.optimizer import remove_unused_nodes
 from onnxscript.rewriter import _fusion_utils, pattern
 
 
@@ -150,8 +148,4 @@ def rewrite(
 gqa_rules = pattern.RewriteRuleSet([_rule1])
 
 
-def fuse_gqa(model: ir.Model, debug: bool = False) -> int:
-    fuse_gqa = _fusion_utils.apply_fusion_rules(gqa_rules)
-    count = fuse_gqa(model, debug=debug)
-    remove_unused_nodes(model)
-    return count
+fuse_gqa = _fusion_utils.apply_fusion_rules(gqa_rules)

onnxscript/rewriter/ort_fusions/skip_normalization.py

@@ -31,9 +31,7 @@ def _skip_rms_normalization(op, input, skip, gamma, epsilon, stash_type):
     return normalized, skip_sum
 
 
-_skip_rms_rule = pattern.RewriteRule(
-    _skip_rms_norm_pattern, _skip_rms_normalization, matcher=pattern.SimplePatternMatcher
-)
+_skip_rms_rule = pattern.RewriteRule(_skip_rms_norm_pattern, _skip_rms_normalization)
 
 skip_rms_normalization_rules = [_skip_rms_rule]
 skip_rms_normalization_ruleset = pattern.RewriteRuleSet(skip_rms_normalization_rules)
@@ -67,9 +65,7 @@ def _skip_layer_normalization(op, input, skip, gamma, beta, epsilon, stash_type)
     return normalized
 
 
-_skip_layer_rule = pattern.RewriteRule(
-    _skip_layer_norm_pattern, _skip_layer_normalization, matcher=pattern.SimplePatternMatcher
-)
+_skip_layer_rule = pattern.RewriteRule(_skip_layer_norm_pattern, _skip_layer_normalization)
 
 skip_layer_normalization_rules = [_skip_layer_rule]
 skip_layer_normalization_ruleset = pattern.RewriteRuleSet(skip_layer_normalization_rules)