From 925a70576e584e77bc930606c59595e9f66b71dd Mon Sep 17 00:00:00 2001
From: Sylvain Jeaugey <sjeaugey@nvidia.com>
Date: Thu, 21 Dec 2017 15:10:09 -0800
Subject: [PATCH] Print NCCL version at start

---
 src/common.cu      | 1 +
 src/nccl1_compat.h | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/src/common.cu b/src/common.cu
index a14c3aa..f47e0f5 100644
--- a/src/common.cu
+++ b/src/common.cu
@@ -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) {
diff --git a/src/nccl1_compat.h b/src/nccl1_compat.h
index 4279789..1a56b57 100644
--- a/src/nccl1_compat.h
+++ b/src/nccl1_compat.h
@@ -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