Size up IPC buffers to multiples of 2MB

Avoid potential CUDA error in concurrent communicator initialization


[ROCm/rccl commit: c8c68fb5f7]
This commit is contained in:
Ke Wen
2019-07-12 08:30:05 -07:00
rodzic ee08e8b421
commit 4211da6d29
2 zmienionych plików z 5 dodań i 3 usunięć
+1 -1
Wyświetl plik
@@ -23,7 +23,7 @@ struct cudaLaunchParams {
#define CACHE_LINE_SIZE 128
#define MEM_ALIGN 4096
#define CUDA_IPC_MIN 2097152UL /* 2MiB - not currently used */
#define CUDA_IPC_MIN 2097152UL
struct ncclSendMem {
union {