Size up IPC buffers to multiples of 2MB

Avoid potential CUDA error in concurrent communicator initialization
此提交包含在:
Ke Wen
2019-07-12 08:30:05 -07:00
父節點 0b192d2299
當前提交 c8c68fb5f7
共有 2 個檔案被更改,包括 5 行新增 和 3 行删除
+1 -1
查看文件
@@ -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 {