typo: fix

shubhambhokare1 · shubhambhokare1 · commit ca15d07f1243 · 2025-03-25T00:00:24.000Z
diff --git a/onnxscript/rewriter/ort_fusions/sdpa.py b/onnxscript/rewriter/ort_fusions/sdpa.py
@@ -84,7 +84,7 @@ def rewrite(self, op, query, key_transposed, value, mask, **_):
     "unmasked_post_div_sdpa", use_mask=False, pre_scale=False, use_mul=False
 )
 unmasked_post_mul_sdpa_rule = SDPA.rule(
-    "unmasked_post_div_sdpa", use_mask=False, pre_scale=False, use_mul=True
+    "unmasked_post_mul_sdpa", use_mask=False, pre_scale=False, use_mul=True
 )
 
 # Rules for SDPA with mask
@@ -98,7 +98,7 @@ def rewrite(self, op, query, key_transposed, value, mask, **_):
     "masked_post_div_sdpa", use_mask=True, pre_scale=False, use_mul=False
 )
 masked_post_mul_sdpa_rule = SDPA.rule(
-    "masked_post_div_sdpa", use_mask=True, pre_scale=False, use_mul=True
+    "masked_post_mul_sdpa", use_mask=True, pre_scale=False, use_mul=True
 )
 
 sdpa_rules = pattern.RewriteRuleSet(

Original file line number	Diff line number	Diff line change
`@@ -84,7 +84,7 @@ def rewrite(self, op, query, key_transposed, value, mask, **_):`
`84`	`84`	`"unmasked_post_div_sdpa", use_mask=False, pre_scale=False, use_mul=False`
`85`	`85`	`)`
`86`	`86`	`unmasked_post_mul_sdpa_rule = SDPA.rule(`
`87`		`- "unmasked_post_div_sdpa", use_mask=False, pre_scale=False, use_mul=True`
	`87`	`+ "unmasked_post_mul_sdpa", use_mask=False, pre_scale=False, use_mul=True`
`88`	`88`	`)`
`89`	`89`
`90`	`90`	`# Rules for SDPA with mask`
`@@ -98,7 +98,7 @@ def rewrite(self, op, query, key_transposed, value, mask, **_):`
`98`	`98`	`"masked_post_div_sdpa", use_mask=True, pre_scale=False, use_mul=False`
`99`	`99`	`)`
`100`	`100`	`masked_post_mul_sdpa_rule = SDPA.rule(`
`101`		`- "masked_post_div_sdpa", use_mask=True, pre_scale=False, use_mul=True`
	`101`	`+ "masked_post_mul_sdpa", use_mask=True, pre_scale=False, use_mul=True`
`102`	`102`	`)`
`103`	`103`
`104`	`104`	`sdpa_rules = pattern.RewriteRuleSet(`