Add support for A100 GPU and related platforms.
Add support for CUDA 11.
Add support for send/receive operations (beta).


[ROCm/rccl commit: 5949d96f36]
Этот коммит содержится в:
Sylvain Jeaugey
2020-05-12 14:40:18 -07:00
родитель c43022b9d8
Коммит 89072c82e5
54 изменённых файлов: 2051 добавлений и 1254 удалений
+6 -2
Просмотреть файл
@@ -1,5 +1,5 @@
/*************************************************************************
* Copyright (c) 2019, NVIDIA CORPORATION. All rights reserved.
* Copyright (c) 2019-2020, NVIDIA CORPORATION. All rights reserved.
*
* See LICENSE.txt for license information
************************************************************************/
@@ -8,7 +8,7 @@
#define NCCL_INFO_H_
#include "nccl.h"
#include "core.h"
#include "devcomm.h"
typedef enum {
ncclPatternRing,
@@ -47,6 +47,10 @@ struct ncclInfo {
size_t nBytes;
int nstepsPerLoop;
int nchunksPerLoop;
ssize_t sendbytes;
ssize_t recvbytes;
uint32_t delta;
int channelId;
};
#endif