Use correct WARP_SIZE for gfx1030 (#348)

Tá an tiomantas seo le fáil i:
Wenkai Du údaraithe agus GitHub tiomanta 2021-04-14 14:09:52 -07:00
1 tuismitheoir a79f74082e
tiomantas 9c718ce6d6
1 comhad athraithe
+4
+4
Féach ar an gComhad
@@ -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