Add support for A100 GPU and related platforms.
Add support for CUDA 11.
Add support for send/receive operations (beta).
Cette révision appartient à :
Sylvain Jeaugey a révisé 2020-06-08 09:31:44 -07:00
1 Parent f36540f55a
révision 5949d96f36
54 fichiers modifiés
+2051 -1254

Aucun fichier ne correspond à votre recherche

+5 -5
Voir le fichier
@@ -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
************************************************************************/
@@ -12,10 +12,10 @@
#include "align.h"
#include <sys/mman.h>
static inline ncclResult_t ncclCudaHostAlloc(void** ptr, void** devPtr, size_t size) {
CUDACHECK(cudaHostAlloc(ptr, size, cudaHostAllocMapped));
memset(*ptr, 0, size);
*devPtr = *ptr;
template <typename T>
static ncclResult_t ncclCudaHostCalloc(T** ptr, size_t nelem) {
CUDACHECK(cudaHostAlloc(ptr, nelem*sizeof(T), cudaHostAllocMapped));
memset(*ptr, 0, nelem*sizeof(T));
return ncclSuccess;
}