kanshan/nccl-tests
1
0
Fork 0
mirror of https://github.com/NVIDIA/nccl-tests.git synced 2026-04-25 08:58:18 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Print NCCL version at start

Browse Source
This commit is contained in:
Sylvain Jeaugey 2017-12-21 15:10:09 -08:00
parent 25016c8eeb
commit 925a70576e
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/common.cu
View File

@ -915,6 +915,7 @@ int main(int argc, char* argv[]) {
NCCLCHECK(ncclGroupEnd());
}
PRINT("# NCCL Tests compiled with NCCL %d.%d\n", NCCL_MAJOR, NCCL_MINOR);
PRINT("# Using devices\n");
for (int p=0; p<nProcs; p++) {
if (p == proc) {

3
src/nccl1_compat.h
View File

@ -8,6 +8,9 @@
#define NCCL1_COMPAT_H
#ifndef NCCL_MAJOR // NCCL 1.x
#define NCCL_MAJOR 1
#define NCCL_MINOR 0
#define ncclNumOps nccl_NUM_OPS
#define ncclNumTypes nccl_NUM_TYPES