修复pd分离开cp引起的校检问题

修改schedule_default和schedule_split_pd的选择

vllm/distributed/kv_transfer/kv_connector/v1/p2p/p2p_nccl_connector.py

@@ -574,7 +574,9 @@ class P2pNcclConnector(KVConnectorBase_V1):
                 num_scheduled_tokens = (
                     scheduler_output.num_scheduled_tokens)[req_id]
                 num_tokens = (num_scheduled_tokens + num_computed_tokens)
-                assert req_id in self.chunked_prefill
+                # assert req_id in self.chunked_prefill
+                if  req_id not in self.chunked_prefill:
+                    continue
                 block_ids = new_block_ids[0]
                 if not resumed_from_preemption:
                     block_ids = (self.chunked_prefill[req_id][0] + block_ids)

vllm/v1/core/sched/scheduler.py

@@ -1024,7 +1024,7 @@ class Scheduler(SchedulerInterface):
         return scheduler_output
     
     def schedule(self) -> SchedulerOutput:
-        if self.num_spec_tokens > 0 or envs.VLLM_USE_PD_SPLIT:
+        if envs.VLLM_USE_PD_SPLIT:
             return self.schedule_split_pd()
         else:
             return self.schedule_default()