DeepSeek/DeepEP
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepEP synced 2025-06-10 00:16:22 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Allow NVLink traffic for low-latency kernels by default

Browse Source
This commit is contained in:
Chenggang Zhao 2025-05-23 20:14:50 +08:00 committed by GitHub
parent 8da1b1f81e
commit aae9fa9a6d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deep_ep/buffer.py
View File

@ -32,7 +32,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 = 12,
allow_nvlink_for_low_latency_mode: bool = False) -> None: allow_nvlink_for_low_latency_mode: bool = True) -> None:
""" """
Initialize the communication buffer. Initialize the communication buffer.