p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update nccl cmake version

This commit is contained in:
ZPaC 2020-10-26 15:24:47 +08:00
parent b4ce0aa933
commit adc48d0c9a
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
cmake/external_libs/nccl.cmake
View File

@ -1,14 +1,14 @@
if (ENABLE_GITEE) if (ENABLE_GITEE)
set(REQ_URL "https://gitee.com/mirrors/nccl/repository/archive/v2.4.8-1.tar.gz") set(REQ_URL "https://gitee.com/mirrors/nccl/repository/archive/v2.7.6-1.tar.gz")
set(MD5 "e3078a91635f6ac12927e9fa5a7248ec") set(MD5 "220d232b30cb9bff2e54219399b9f6fb")
else() else()
set(REQ_URL "https://github.com/NVIDIA/nccl/archive/v2.4.8-1.tar.gz") set(REQ_URL "https://github.com/NVIDIA/nccl/archive/v2.7.6-1.tar.gz")
set(MD5 "f14b37d6af1c79db5f57cb029a753727") set(MD5 "073b19899f374c5ba07d2db02dc38f9f")
endif () endif ()
set(nccl_CFLAGS "-D_FORTIFY_SOURCE=2 -O2") set(nccl_CFLAGS "-D_FORTIFY_SOURCE=2 -O2")
mindspore_add_pkg(nccl mindspore_add_pkg(nccl
VER 2.4.8-1 VER 2.7.6-1
LIBS nccl LIBS nccl
URL ${REQ_URL} URL ${REQ_URL}
MD5 ${MD5} MD5 ${MD5}