From 1addf4f196d574fe3dfc0152e111b0fcd3041086 Mon Sep 17 00:00:00 2001
From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com>
Date: Tue, 7 Jul 2020 13:30:39 -0700
Subject: [PATCH] Match RCCL package name to API version (#229)

---
 CMakeLists.txt | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index fa42e4c6cd..96a0bd9f77 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -62,7 +62,11 @@ else()
 endif()
 
 # Setup VERSION
-rocm_setup_version(VERSION "2.7.0")
+rocm_get_commit_count(COMMIT_COUNT)
+rocm_get_version(OUTPUT_VERSION)
+rocm_version_regex_parse("^[0-9]+\\.[0-9]+\\.[0-9]+[.-][0-9]+[.-](.*)" COMMIT_HASH "${OUTPUT_VERSION}")
+set(VERSION "${NCCL_MAJOR}.${NCCL_MINOR}.${NCCL_PATCH}-${COMMIT_COUNT}-${COMMIT_HASH}")
+rocm_setup_version(NO_GIT_TAG_VERSION VERSION "${NCCL_MAJOR}.${NCCL_MINOR}.${NCCL_PATCH}-${COMMIT_COUNT}-${COMMIT_HASH}")
 
 list(APPEND CMAKE_PREFIX_PATH
             /opt/rocm