AI/rocm-systems
2
0
Förgrening 0
Kod Ärenden Pull-förfrågningar Actions Packages Projekt Släpp Wiki Aktiviteter

Use correct WARP_SIZE for gfx1030 (#348)

Bläddra i källkod 
[ROCm/rccl commit: 9c718ce6d6]
This commit is contained in:
Wenkai Du
2021-04-14 14:09:52 -07:00
committad av GitHub
förälder 661b1351a3
incheckning 951d89b12f
1 ändrade filer med 4 tillägg och 0 borttagningar
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/src/include/devcomm.h
+4
Visa fil
@@ -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