Fix normalized key

gramalingam · gramalingam · commit f1053eacb00c · 2025-10-17T14:57:42.000-07:00
diff --git a/onnxscript/rewriter/ort_fusions/gqa_test.py b/onnxscript/rewriter/ort_fusions/gqa_test.py
@@ -484,7 +484,7 @@ def gqa(query, key, value, past_key, past_value, cos, sin, query_scale, key_scal
                 sin,
             )
             key_BHkvSDh_rope = msft_op.RotaryEmbedding(
-                key_BHkvSDh,
+                key_BHkvSDh_normalized,
                 position_ids_k,
                 cos,
                 sin,

Original file line number	Diff line number	Diff line change
`@@ -484,7 +484,7 @@ def gqa(query, key, value, past_key, past_value, cos, sin, query_scale, key_scal`
`484`	`484`	`sin,`
`485`	`485`	`)`
`486`	`486`	`key_BHkvSDh_rope = msft_op.RotaryEmbedding(`
`487`		`- key_BHkvSDh,`
	`487`	`+ key_BHkvSDh_normalized,`
`488`	`488`	`position_ids_k,`
`489`	`489`	`cos,`
`490`	`490`	`sin,`