From 7f9c15b8430e054ccadc2d6cc09e7bfdfa3fd2ca 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)

[ROCm/rccl commit: 41e47a36e706597aa4271ca3243964b5adefde65]
---
 projects/rccl/src/clique/AllReduceCliqueKernel.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/projects/rccl/src/clique/AllReduceCliqueKernel.h b/projects/rccl/src/clique/AllReduceCliqueKernel.h
index 67a306d3c7..9889e9a9d5 100644
--- a/projects/rccl/src/clique/AllReduceCliqueKernel.h
+++ b/projects/rccl/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<ALL_REDUCE_CLIQUE_UNROLL, FUNC, T, NUM_RANKS, NUM_RANKS, NUM_RANKS, NUM_RANKS>(
       threadIdx.x, blockDim.x, NUM_RANKS, srcs, NUM_RANKS, dsts, blockN);
   }