From 574f0aca5342104288d3bc0db98b8ee4a3fa58e0 Mon Sep 17 00:00:00 2001
From: Wenkai Du <wenkai.du@amd.com>
Date: Thu, 26 Aug 2021 11:10:58 -0700
Subject: [PATCH] Fix typo that affects clique kernels

---
 src/collectives/device/common.h | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/collectives/device/common.h b/src/collectives/device/common.h
index fef1df9d21..c0fabcae3d 100644
--- a/src/collectives/device/common.h
+++ b/src/collectives/device/common.h
@@ -92,7 +92,7 @@
   NCCL_FUNCS3C(func, Prod), \
   NCCL_FUNCS3C(func, Max ), \
   NCCL_FUNCS3C(func, Min ), \
-  NCCL_FUNCS3A(func, Avg)
+  NCCL_FUNCS3C(func, Avg)
 
 // Must be consistent with ncclFunc_t
 #define NCCL_FUNCS() { \
@@ -116,7 +116,7 @@ static const __device__ constexpr ncclKernelFunc_t ncclFuncs[]{
   NCCL_FUNCS2A(Reduce),
   NCCL_FUNCS2B(AllGather),
   NCCL_FUNCS2A(ReduceScatter),
-  NCCL_FUNCS2A(AllReduce),
+  NCCL_FUNCS2C(AllReduce),
   NCCL_FUNC_NAME(SendRecv, RING, SIMPLE, Sum, int8_t),
 #endif
 };