fix rope acc error

b7b4e639 · zhuwenwen · 9e5ec696 · b7b4e639
Commit b7b4e639 authored Oct 05, 2025 by zhuwenwen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

vllm/model_executor/layers/rotary_embedding.py vllm/model_executor/layers/rotary_embedding.py +1 -1

No files found.
--- a/vllm/model_executor/layers/rotary_embedding.py
+++ b/vllm/model_executor/layers/rotary_embedding.py
@@ -916,7 +916,7 @@ class DeepseekScalingRotaryEmbedding(RotaryEmbedding):
    direct_register_custom_op(
        op_name="rotary_embedding_deepseek_fuse",
        op_func=rotary_embedding_deepseek_fuse,
-        mutates_args=[], 
+        mutates_args=["query", "key"], 
        fake_impl=rotary_embedding_deepseek_fuse_fake,
    )