Match RCCL package name to API version (#229)

Этот коммит содержится в:
Wenkai Du
2020-07-07 13:30:39 -07:00
коммит произвёл GitHub
родитель 9bd4c14603
Коммит 1addf4f196
+5 -1
Просмотреть файл
@@ -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