Size up IPC buffers to multiples of 2MB

Avoid potential CUDA error in concurrent communicator initialization
This commit is contained in:
Ke Wen
2019-07-12 08:30:05 -07:00
parent 0b192d2299
commit c8c68fb5f7
2 ha cambiato i file con 5 aggiunte e 3 eliminazioni
+1 -1
Vedi File
@@ -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 {