Rename fusion files (#2476)

Rename fusion files to follow a uniform style and to locate them easily:

* fuse_packed_qkv_gqa => gqa_packed_qkv
* fuse_mha_bias => mha_bias
* Rename corresponding test file also

---------

Signed-off-by: Ganesan Ramalingam <[email protected]>

Author: gramalingam

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,

onnxscript/rewriter/ort_fusions/fuse_packed_qkv_gqa.py renamed to onnxscript/rewriter/ort_fusions/gqa_packed_qkv.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)
 

onnxscript/rewriter/ort_fusions/fuse_packed_qkv_gqa_test.py renamed to onnxscript/rewriter/ort_fusions/gqa_packed_qkv_test.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)