obscura/nccl-tests
1
0
Fork 0
mirror of https://github.com/NVIDIA/nccl-tests.git synced 2026-10-01 14:37:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'github/master'

Browse Source 
NCCL_TESTS_VERSION 2.17.10
This commit is contained in:
David Addison
2026-03-05 16:24:48 -08:00
parent cd9cad3de4 c379e19a71
commit 42b9e47067
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/common.cu
+4 -2
View File
@@ -1342,8 +1342,10 @@ testResult_t run() {
#if NCCL_VERSION_CODE >= NCCL_VERSION(2,28,0)
ncclDevComm devComms[nThreads*nGpus];
#endif
int64_t initGpuMem[nThreads] = {0};
int64_t bufferMemory[nThreads] = {0};
int64_t initGpuMem[nThreads];
int64_t bufferMemory[nThreads];
memset(initGpuMem, 0, sizeof(initGpuMem));
memset(bufferMemory, 0, sizeof(bufferMemory));
if (!parallel_init) {
// Capture the memory used by the GPUs before initializing the NCCL communicators
int64_t* initFreeGpuMem = (int64_t*)calloc(nGpus*3, sizeof(int64_t));
src/common.h
+1 -1
View File
@@ -6,7 +6,7 @@
#ifndef __COMMON_H__
#define __COMMON_H__
#define NCCL_TESTS_VERSION "2.17.9"
#define NCCL_TESTS_VERSION "2.17.10"
#include "nccl.h"
#if NCCL_VERSION_CODE >= NCCL_VERSION(2,28,0)