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
bovenliggende 0b192d2299
commit c8c68fb5f7
2 gewijzigde bestanden met toevoegingen van 5 en 3 verwijderingen
+1 -1
Bestand weergeven
@@ -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 {