typos

Author: shubhambhokare1

onnxscript/rewriter/ort_fusions/sdpa_test.py

@@ -197,7 +197,7 @@ def _custom_scale_pre_div_sdpa_script(query, key, value, mask):
 
 
 @script()
-def _custom_scale_mul_sdpa_script(query, key, value, mask):
+def _custom_scale_pre_mul_sdpa_script(query, key, value, mask):
     key_transposed = op.Transpose(key, perm=[0, 1, 3, 2])
     multiplier = op.Constant(value_float=0.5)
     scaled_query = op.Mul(query, multiplier)
@@ -278,7 +278,7 @@ class TestSDPAFusion(unittest.TestCase):
             ("custom_scale_post_div_masked", _custom_scale_post_div_sdpa_script),
             ("custom_scale_pre_mul_masked", _custom_scale_pre_mul_sdpa_script),
             ("custom_scale_pre_div_masked", _custom_scale_pre_div_sdpa_script),
-            (_custom_multi_scale_pre_mul_sdpa_script, _custom_multi_scale_pre_mul_sdpa_script),
+            ("_custom_multi_scale_pre_mul_sdpa_script", _custom_multi_scale_pre_mul_sdpa_script),
         ]
     )
     def test_sdpa_fusion(self, name, script_func):