From d93915ce6ae9070a41496a8d7609a76c5a99b6f3 Mon Sep 17 00:00:00 2001 From: Ganesan Ramalingam Date: Wed, 27 Aug 2025 17:33:12 -0700 Subject: [PATCH] Remove function extraction in ONNX rotary embedding Signed-off-by: Ganesan Ramalingam --- onnxscript/rewriter/onnx_fusions/_rotary_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxscript/rewriter/onnx_fusions/_rotary_embedding.py b/onnxscript/rewriter/onnx_fusions/_rotary_embedding.py index 55620a7b41..2009c6953f 100644 --- a/onnxscript/rewriter/onnx_fusions/_rotary_embedding.py +++ b/onnxscript/rewriter/onnx_fusions/_rotary_embedding.py @@ -30,7 +30,7 @@ def _rotate_half_pattern(op, x, start1, end1, start2, end2): class RotaryEmbedding23Fusion(pattern.RewriteRuleClassBase): def __init__(self): - super().__init__(name="RotaryEmbedding23", as_function=True) + super().__init__(name="RotaryEmbedding23") def pattern(self, op, x, cos, sin, start1, end1, start2, end2): return x * cos + _rotate_half_pattern(op, x, start1, end1, start2, end2) * sin