Use correct WARP_SIZE for gfx1030 (#348)

2021-04-14 14:09:52 -07:00
parent a79f74082e
commit 9c718ce6d6
1 changed files with 4 additions and 0 deletions
@@ -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