Print NCCL version at start

Tento commit je obsažen v:
Sylvain Jeaugey
2017-12-21 15:10:09 -08:00
rodič 25016c8eeb
revize 925a70576e
2 změnil soubory, kde provedl 4 přidání a 0 odebrání
+1
Zobrazit soubor
@@ -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
Zobrazit soubor
@@ -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