File tree Expand file tree Collapse file tree 1 file changed +1
-5
lines changed Expand file tree Collapse file tree 1 file changed +1
-5
lines changed Original file line number Diff line number Diff line change @@ -433,12 +433,8 @@ def graph_optimize_and_warm_up_model(self) -> None:
433
433
self .worker .graph_optimize_and_warm_up_model ()
434
434
# reset cache_messager prefilled_step signal
435
435
if self .scheduler_config .splitwise_role == "prefill" :
436
- dp_rank_id = (
437
- self .local_rank
438
- + self .parallel_config .local_data_parallel_id * self .parallel_config .tensor_parallel_size
439
- )
440
436
gpu_id = self .worker .model_runner .device_id
441
- prefilled_step_name = f"splitwise_complete_prefilled_step_{ dp_rank_id } "
437
+ prefilled_step_name = f"splitwise_complete_prefilled_step_{ self . local_rank } "
442
438
prefilled_step_idx_data = np .zeros (shape = [1 ], dtype = np .int32 )
443
439
step_shm_value = IPCSignal (
444
440
name = prefilled_step_name ,
You can’t perform that action at this time.
0 commit comments