Unverified Commit aae9fa9a authored by Chenggang Zhao's avatar Chenggang Zhao Committed by GitHub
Browse files

Allow NVLink traffic for low-latency kernels by default

parent 8da1b1f8
...@@ -32,7 +32,7 @@ class Buffer: ...@@ -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.
......
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