From 41e47a36e706597aa4271ca3243964b5adefde65 Mon Sep 17 00:00:00 2001 From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com> Date: Fri, 15 Jan 2021 17:48:10 -0800 Subject: [PATCH] Use less unroll for clique kernels (#313) --- src/clique/AllReduceCliqueKernel.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/clique/AllReduceCliqueKernel.h b/src/clique/AllReduceCliqueKernel.h index 67a306d3c7..9889e9a9d5 100644 --- a/src/clique/AllReduceCliqueKernel.h +++ b/src/clique/AllReduceCliqueKernel.h @@ -67,7 +67,7 @@ __device__ void AllReduceCliqueSplitKernel(struct CollectiveArgs* args) } // Perform the reduction - #define ALL_REDUCE_CLIQUE_UNROLL 2 + #define ALL_REDUCE_CLIQUE_UNROLL 1 ReduceOrCopyMulti( threadIdx.x, blockDim.x, NUM_RANKS, srcs, NUM_RANKS, dsts, blockN); }