[PD] Fix Assertion failed: /DeepEP/csrc/kernels/internode.cu:483, condition: ibgda_get_state()->num_rc_per_pe >= num_channels #134 (#5830)
This commit is contained in:
@@ -84,9 +84,7 @@ class DeepEPBuffer:
|
|||||||
num_nvl_bytes,
|
num_nvl_bytes,
|
||||||
num_rdma_bytes,
|
num_rdma_bytes,
|
||||||
low_latency_mode=deepep_mode.enable_low_latency(),
|
low_latency_mode=deepep_mode.enable_low_latency(),
|
||||||
num_qps_per_rank=(
|
num_qps_per_rank=(max(num_experts // group.size(), Buffer.num_sms // 2)),
|
||||||
num_experts // group.size() if deepep_mode.enable_low_latency() else 1
|
|
||||||
),
|
|
||||||
)
|
)
|
||||||
return cls._buffer
|
return cls._buffer
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user