Unverified Commit cc4a80ca authored by ybyang's avatar ybyang Committed by GitHub
Browse files

[PD] Fix Assertion failed: /DeepEP/csrc/kernels/internode.cu:483, condition:...

[PD] Fix Assertion failed: /DeepEP/csrc/kernels/internode.cu:483, condition: ibgda_get_state()->num_rc_per_pe >= num_channels #134 (#5830)
parent 3c8a5231
......@@ -84,9 +84,7 @@ class DeepEPBuffer:
num_nvl_bytes,
num_rdma_bytes,
low_latency_mode=deepep_mode.enable_low_latency(),
num_qps_per_rank=(
num_experts // group.size() if deepep_mode.enable_low_latency() else 1
),
num_qps_per_rank=(max(num_experts // group.size(), Buffer.num_sms // 2)),
)
return cls._buffer
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment