From a8511ca628f05cb83ed03e0b196ba89e6f710d57 Mon Sep 17 00:00:00 2001 From: Michael Axtmann Date: Fri, 13 Jan 2023 08:10:51 -0800 Subject: [PATCH] Change long lat defaults. --- src/enqueue.cc | 4 ++-- src/init.cc | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/enqueue.cc b/src/enqueue.cc index 413b337..9fa7d73 100644 --- a/src/enqueue.cc +++ b/src/enqueue.cc @@ -1268,8 +1268,8 @@ NCCL_PARAM(PipelineChunkSteps, "PIPELINE_CHUNKSTEPS", (NCCL_STEPS/8)); NCCL_PARAM(RingChunkSteps, "RING_CHUNKSTEPS", (NCCL_STEPS/2)); NCCL_PARAM(RingSliceSteps, "RING_SLICESTEPS", (NCCL_STEPS/4)); -NCCL_PARAM(TreeLongNetLatChunkSteps, "TREE_LONG_NET_LAT_CHUNKSTEPS", (4)); -NCCL_PARAM(TreeLongNetLatSliceSteps, "TREE_LONG_NET_LAT_SLICESTEPS", (2)); +NCCL_PARAM(TreeLongNetLatChunkSteps, "TREE_LONG_NET_LAT_CHUNKSTEPS", (1)); +NCCL_PARAM(TreeLongNetLatSliceSteps, "TREE_LONG_NET_LAT_SLICESTEPS", (1)); NCCL_PARAM(RingLongNetLatChunkSteps, "RING_LONG_NET_LAT_CHUNKSTEPS", (4)); NCCL_PARAM(RingLongNetLatSliceSteps, "RING_LONG_NET_LAT_SLICESTEPS", (2)); diff --git a/src/init.cc b/src/init.cc index 8521008..845e870 100644 --- a/src/init.cc +++ b/src/init.cc @@ -470,7 +470,7 @@ static ncclResult_t setupChannel(struct ncclComm* comm, int channelId, int rank, NCCL_PARAM(BuffSize, "BUFFSIZE", -2); NCCL_PARAM(LlBuffSize, "LL_BUFFSIZE", -2); NCCL_PARAM(Ll128BuffSize, "LL128_BUFFSIZE", -2); -NCCL_PARAM(LongNetLatBuffSizeScaling, "LONG_NET_LAT_BUFFSIZE_SCALING", 4); +NCCL_PARAM(LongNetLatBuffSizeScaling, "LONG_NET_LAT_BUFFSIZE_SCALING", 1); NCCL_PARAM(P2pNetChunkSize, "P2P_NET_CHUNKSIZE", (1 << 17)); /* 128 kB */ NCCL_PARAM(P2pPciChunkSize, "P2P_PCI_CHUNKSIZE", (1 << 17)); /* 128 kB */