Merge pull request !7514 from ZPaC/master-download-from-giteetags/v1.1.0
| @@ -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 | |||||
| MD5 f14b37d6af1c79db5f57cb029a753727 | |||||
| URL ${REQ_URL} | |||||
| 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/*) | ||||
| @@ -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 | |||||
| MD5 393c0e27b68bfaf96718caa3aa96f5a3 | |||||
| URL ${REQ_URL} | |||||
| 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/* | ||||