modify rule names

shubhambhokare1 · shubhambhokare1 · commit d82adcb6d26f · 2025-05-01T07:10:53.000Z
diff --git a/onnxscript/rewriter/ort_fusions/skip_normalization.py b/onnxscript/rewriter/ort_fusions/skip_normalization.py
@@ -93,7 +93,7 @@ def rewrite(self, op, input, skip, gamma, bias, epsilon, stash_type):
     "SkipRmsNormBias", has_bias=True, bias_pre_add=False
 )
 _skip_rms_pre_add_bias_rule = SkipRmsNormFusion.rule(
-    "SkipRmsNormBias", has_bias=True, bias_pre_add=True
+    "SkipRmsNormPreBias", has_bias=True, bias_pre_add=True
 )
 _skip_rms_rule = SkipRmsNormFusion.rule("SkipRmsNorm", has_bias=False)
 
@@ -184,7 +184,7 @@ def rewrite(self, op, input, skip, gamma, beta, bias, epsilon, stash_type):
     "SkipLayerNormBias", has_bias=True, bias_pre_add=False
 )
 _skip_layer_pre_add_bias_rule = SkipLayerNormFusion.rule(
-    "SkipLayerNormBias", has_bias=True, bias_pre_add=True
+    "SkipLayerNormPreBias", has_bias=True, bias_pre_add=True
 )
 _skip_layer_rule = SkipLayerNormFusion.rule("SkipLayerNorm", has_bias=False)
 

Original file line number	Diff line number	Diff line change
`@@ -93,7 +93,7 @@ def rewrite(self, op, input, skip, gamma, bias, epsilon, stash_type):`
`93`	`93`	`"SkipRmsNormBias", has_bias=True, bias_pre_add=False`
`94`	`94`	`)`
`95`	`95`	`_skip_rms_pre_add_bias_rule = SkipRmsNormFusion.rule(`
`96`		`- "SkipRmsNormBias", has_bias=True, bias_pre_add=True`
	`96`	`+ "SkipRmsNormPreBias", has_bias=True, bias_pre_add=True`
`97`	`97`	`)`
`98`	`98`	`_skip_rms_rule = SkipRmsNormFusion.rule("SkipRmsNorm", has_bias=False)`
`99`	`99`
`@@ -184,7 +184,7 @@ def rewrite(self, op, input, skip, gamma, beta, bias, epsilon, stash_type):`
`184`	`184`	`"SkipLayerNormBias", has_bias=True, bias_pre_add=False`
`185`	`185`	`)`
`186`	`186`	`_skip_layer_pre_add_bias_rule = SkipLayerNormFusion.rule(`
`187`		`- "SkipLayerNormBias", has_bias=True, bias_pre_add=True`
	`187`	`+ "SkipLayerNormPreBias", has_bias=True, bias_pre_add=True`
`188`	`188`	`)`
`189`	`189`	`_skip_layer_rule = SkipLayerNormFusion.rule("SkipLayerNorm", has_bias=False)`
`190`	`190`