Print NCCL version at start

[ROCm/rccl-tests commit: 925a70576e]
Cette révision appartient à :
Sylvain Jeaugey
2017-12-21 15:10:09 -08:00
Parent 224eddfc15
révision 3958cc347e
2 fichiers modifiés avec 4 ajouts et 0 suppressions
+1
Voir le fichier
@@ -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
Voir le fichier
@@ -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