From 951d89b12ffec2f91103addbd76cff68c826921c Mon Sep 17 00:00:00 2001
From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com>
Date: Wed, 14 Apr 2021 14:09:52 -0700
Subject: [PATCH] Use correct WARP_SIZE for gfx1030 (#348)

[ROCm/rccl commit: 9c718ce6d6f620be4084b33f5cb45046a8c78386]
---
 projects/rccl/src/include/devcomm.h | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/projects/rccl/src/include/devcomm.h b/projects/rccl/src/include/devcomm.h
index 9997de7f7c..117dc40d74 100644
--- a/projects/rccl/src/include/devcomm.h
+++ b/projects/rccl/src/include/devcomm.h
@@ -61,7 +61,11 @@ union ncclLLFifoLine {
   int4 i4;
 };
 
+#if defined(__gfx1030__)
+#define WARP_SIZE 32
+#else
 #define WARP_SIZE 64
+#endif
 #define MAXCHANNELS 32
 #define NCCL_MAX_NTHREADS 256
 #define NCCL_SIMPLE_MAX_NTHREADS NCCL_MAX_NTHREADS