Skip to content

Commit 2ff01f7

Browse files
authored
Remove function extraction in ONNX rotary embedding (#2525)
Remove as_function attribute in ONNX rotary embedding fusion. (It is not needed since it is a standard op.) Signed-off-by: Ganesan Ramalingam <[email protected]>
1 parent 0766199 commit 2ff01f7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

onnxscript/rewriter/onnx_fusions/_rotary_embedding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def _rotate_half_pattern(op, x, start1, end1, start2, end2):
3030

3131
class RotaryEmbedding23Fusion(pattern.RewriteRuleClassBase):
3232
def __init__(self):
33-
super().__init__(name="RotaryEmbedding23", as_function=True)
33+
super().__init__(name="RotaryEmbedding23")
3434

3535
def pattern(self, op, x, cos, sin, start1, end1, start2, end2):
3636
return x * cos + _rotate_half_pattern(op, x, start1, end1, start2, end2) * sin

0 commit comments

Comments
 (0)