[fix]修复不开pcp时打开mtp报错

e7dcfb5b · 王敏 · 561b6cbb · e7dcfb5b
Commit e7dcfb5b authored Apr 11, 2026 by 王敏
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

vllm/v1/spec_decode/eagle.py vllm/v1/spec_decode/eagle.py +2 -2

No files found.
--- a/vllm/v1/spec_decode/eagle.py
+++ b/vllm/v1/spec_decode/eagle.py
@@ -222,9 +222,9 @@ class SpecDecodeBaseProposer:
            1, len(self.tree_choices) + 1, device=device, dtype=torch.int32
        ).repeat(max_batch_size, 1)

+        self.scatter_indexes_tensor = None
+        self.gather_indexes_tensor = None
        if envs.VLLM_MLA_CP:
-            self.scatter_indexes_tensor = None
-            self.gather_indexes_tensor = None
            self.query_start_loc = CpuGpuBuffer(
                max_batch_size + 1,
                dtype=torch.int32,