From 3958cc347e8f0246cbcbde20f01333e1fc6125c6 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

[ROCm/rccl-tests commit: 925a70576e584e77bc930606c59595e9f66b71dd]
---
 projects/rccl-tests/src/common.cu      | 1 +
 projects/rccl-tests/src/nccl1_compat.h | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/projects/rccl-tests/src/common.cu b/projects/rccl-tests/src/common.cu
index a14c3aac01..f47e0f5da5 100644
--- a/projects/rccl-tests/src/common.cu
+++ b/projects/rccl-tests/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/projects/rccl-tests/src/nccl1_compat.h b/projects/rccl-tests/src/nccl1_compat.h
index 4279789af6..1a56b571cf 100644
--- a/projects/rccl-tests/src/nccl1_compat.h
+++ b/projects/rccl-tests/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