Rename fusion files (#2476)

gramalingam · web-flow · commit 32f21967d848 · 2025-08-05T17:15:23.000-07:00
Rename fusion files to follow a uniform style and to locate them easily:

* fuse_packed_qkv_gqa =&gt; gqa_packed_qkv
* fuse_mha_bias =&gt; mha_bias
* Rename corresponding test file also

---------

Signed-off-by: Ganesan Ramalingam &lt;grama@microsoft.com&gt;
diff --git a/onnxscript/rewriter/ort_fusions/_core.py b/onnxscript/rewriter/ort_fusions/_core.py
@@ -17,11 +17,11 @@
 from onnxscript.rewriter.ort_fusions.bias_gelu import fuse_bias_gelu
 from onnxscript.rewriter.ort_fusions.cos_sin_cache import fuse_cos_sin_cache
 from onnxscript.rewriter.ort_fusions.erfgelu import fuse_erfgelu
-from onnxscript.rewriter.ort_fusions.fuse_mha_bias import fuse_mha_bias
-from onnxscript.rewriter.ort_fusions.fuse_packed_qkv_gqa import fuse_qkv_gqa
 from onnxscript.rewriter.ort_fusions.gelu import fuse_gelu
 from onnxscript.rewriter.ort_fusions.gqa import fuse_gqa
+from onnxscript.rewriter.ort_fusions.gqa_packed_qkv import fuse_qkv_gqa
 from onnxscript.rewriter.ort_fusions.mha import fuse_mha1, fuse_mha2
+from onnxscript.rewriter.ort_fusions.mha_bias import fuse_mha_bias
 from onnxscript.rewriter.ort_fusions.rms_normalization import fuse_rms_normalization
 from onnxscript.rewriter.ort_fusions.rotary_embedding import (
     fuse_partial_rotary_embedding,
diff --git a/onnxscript/rewriter/ort_fusions/gqa_packed_qkv.py b/onnxscript/rewriter/ort_fusions/gqa_packed_qkv.py
diff --git a/onnxscript/rewriter/ort_fusions/gqa_packed_qkv_test.py b/onnxscript/rewriter/ort_fusions/gqa_packed_qkv_test.py
@@ -14,7 +14,7 @@
 from onnxscript import FLOAT, INT32, script
 from onnxscript import opset18 as op
 from onnxscript.rewriter.ort_fusions._test_utils import assert_allclose
-from onnxscript.rewriter.ort_fusions.fuse_packed_qkv_gqa import fuse_qkv_gqa
+from onnxscript.rewriter.ort_fusions.gqa_packed_qkv import fuse_qkv_gqa
 
 msft_op = onnxscript.values.Opset("com.microsoft", 1)
 
diff --git a/onnxscript/rewriter/ort_fusions/mha_bias.py b/onnxscript/rewriter/ort_fusions/mha_bias.py
@@ -163,7 +163,7 @@ def rewrite(
         )
 
 
-fuse_mha_bias_rules = pattern.RewriteRuleSet([FuseBiasMHA.rule()])
+mha_bias_rules = pattern.RewriteRuleSet([FuseBiasMHA.rule()])
 
 
-fuse_mha_bias = _fusion_utils.apply_fusion_rules(fuse_mha_bias_rules)
+fuse_mha_bias = _fusion_utils.apply_fusion_rules(mha_bias_rules)

Original file line number	Diff line number	Diff line change
`@@ -163,7 +163,7 @@ def rewrite(`
`163`	`163`	`)`
`164`	`164`
`165`	`165`
`166`		`-fuse_mha_bias_rules = pattern.RewriteRuleSet([FuseBiasMHA.rule()])`
	`166`	`+mha_bias_rules = pattern.RewriteRuleSet([FuseBiasMHA.rule()])`
`167`	`167`
`168`	`168`
`169`		`-fuse_mha_bias = _fusion_utils.apply_fusion_rules(fuse_mha_bias_rules)`
	`169`	`+fuse_mha_bias = _fusion_utils.apply_fusion_rules(mha_bias_rules)`