[Fix]修复PP模式pcp报错

vllm/model_executor/models/deepseek_v2.py

@@ -1409,6 +1409,9 @@ class DeepseekV2Model(nn.Module):
             )
 
         if not get_pp_group().is_last_rank:
+            if enable_mla_cp:
+                hidden_states = tensor_model_parallel_all_gather(hidden_states.contiguous(), dim=0)
+                residual = tensor_model_parallel_all_gather(residual.contiguous(), dim=0)
             return IntermediateTensors(
                 {"hidden_states": hidden_states, "residual": residual}
             )