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

download nccl and pslite from gitee

This commit is contained in:
ZPaC 2020-10-20 17:21:04 +08:00
parent 15f57029a5
commit d07c6151ea
2 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
cmake/external_libs/nccl.cmake
View File

@ -1,10 +1,17 @@
if (ENABLE_GITEE)
set(REQ_URL "https://gitee.com/mirrors/nccl/repository/archive/v2.4.8-1.tar.gz")
set(MD5 "e3078a91635f6ac12927e9fa5a7248ec")
else()
set(REQ_URL "https://github.com/NVIDIA/nccl/archive/v2.4.8-1.tar.gz")
set(MD5 "f14b37d6af1c79db5f57cb029a753727")
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.4.8-1
LIBS nccl LIBS nccl
URL https://github.com/NVIDIA/nccl/archive/v2.4.8-1.tar.gz URL ${REQ_URL}
MD5 f14b37d6af1c79db5f57cb029a753727 MD5 ${MD5}
BUILD_OPTION src.build NVCC_GENCODE="-gencode=arch=compute_70,code=sm_70" BUILD_OPTION src.build NVCC_GENCODE="-gencode=arch=compute_70,code=sm_70"
INSTALL_INCS build/include/* INSTALL_INCS build/include/*
INSTALL_LIBS build/lib/*) INSTALL_LIBS build/lib/*)

12
cmake/external_libs/pslite.cmake
View File

@ -1,11 +1,19 @@
if (ENABLE_GITEE)
set(REQ_URL "https://gitee.com/mirrors/ps-lite/repository/archive/34fd45cae457d59850fdcb2066467778d0673f21.zip")
set(MD5 "0d1543b8dcb0bc3610637e1643c94eb4")
else()
set(REQ_URL "https://github.com/dmlc/ps-lite/archive/34fd45cae457d59850fdcb2066467778d0673f21.zip")
set(MD5 "393c0e27b68bfaf96718caa3aa96f5a3")
endif ()
set(pslite_USE_STATIC_LIBS ON) set(pslite_USE_STATIC_LIBS ON)
if (${ENABLE_IBVERBS} STREQUAL "ON") if (${ENABLE_IBVERBS} STREQUAL "ON")
set(pslite_CXXFLAGS "USE_IBVERBS=1") set(pslite_CXXFLAGS "USE_IBVERBS=1")
endif() endif()
mindspore_add_pkg(pslite mindspore_add_pkg(pslite
LIBS ps LIBS ps
URL https://github.com/dmlc/ps-lite/archive/34fd45cae457d59850fdcb2066467778d0673f21.zip URL ${REQ_URL}
MD5 393c0e27b68bfaf96718caa3aa96f5a3 MD5 ${MD5}
PATCHES ${CMAKE_SOURCE_DIR}/third_party/patch/pslite/ps_lite.patch001 PATCHES ${CMAKE_SOURCE_DIR}/third_party/patch/pslite/ps_lite.patch001
ONLY_MAKE True ONLY_MAKE True
ONLY_MAKE_INCS include/* ONLY_MAKE_INCS include/*