Skip to content

GitLab

Unverified Commit 4e72eb39 authored by Shangyan Zhou's avatar Shangyan Zhou Committed by GitHub
Browse files

Change the default num of QPs. (#263)

parent 58c47942
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
deep_ep/buffer.py
View file @ 4e72eb39
...@@ -31,7 +31,7 @@ class Buffer: ...@@ -31,7 +31,7 @@ class Buffer:
def __init__(self, group: dist.ProcessGroup, def __init__(self, group: dist.ProcessGroup,
num_nvl_bytes: int = 0, num_rdma_bytes: int = 0, num_nvl_bytes: int = 0, num_rdma_bytes: int = 0,
low_latency_mode: bool = False, num_qps_per_rank: int = 12, low_latency_mode: bool = False, num_qps_per_rank: int = 24,
allow_nvlink_for_low_latency_mode: bool = True, allow_nvlink_for_low_latency_mode: bool = True,
allow_mnnvl: bool = False) -> None: allow_mnnvl: bool = False) -> None:
""" """
......
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