[XPU] refactor of block_attn param 'pos_emb_type' (#5511)

2025-12-24 13:28:13 +08:00 · 2025-12-12 14:30:09 +08:00
parent 4eb55332f6
commit 888c4b992d
6 changed files with 25 additions and 19 deletions
--- a/fastdeploy/model_executor/layers/attention/xpu_attn_backend.py
+++ b/fastdeploy/model_executor/layers/attention/xpu_attn_backend.py
@@ -213,7 +213,7 @@ class XPUAttentionBackend(AttentionBackend):
            None,  # smooth
            metadata.kv_signal_data_list[layer.layer_id],  # kv_signal_data
            forward_meta.kv_signal_sender,  # kv_signal_sender
-            forward_meta.pos_emb_type,
+            layer.use_neox_rotary_style,
            self.rope_3d,
        )