From d07c6151ea218fd9cd129a7a5066660a11abf938 Mon Sep 17 00:00:00 2001
From: ZPaC <zhoupeichen@huawei.com>
Date: Tue, 20 Oct 2020 17:21:04 +0800
Subject: [PATCH] download nccl and pslite from gitee

---
 cmake/external_libs/nccl.cmake   | 11 +++++++++--
 cmake/external_libs/pslite.cmake | 12 ++++++++++--
 2 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/cmake/external_libs/nccl.cmake b/cmake/external_libs/nccl.cmake
index b74946f7fb..aea9c6f0fd 100644
--- a/cmake/external_libs/nccl.cmake
+++ b/cmake/external_libs/nccl.cmake
@@ -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")
 mindspore_add_pkg(nccl
         VER 2.4.8-1
         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"
         INSTALL_INCS build/include/*
         INSTALL_LIBS build/lib/*)
diff --git a/cmake/external_libs/pslite.cmake b/cmake/external_libs/pslite.cmake
index 28c851b094..05e3205645 100644
--- a/cmake/external_libs/pslite.cmake
+++ b/cmake/external_libs/pslite.cmake
@@ -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)
 if (${ENABLE_IBVERBS} STREQUAL "ON")
     set(pslite_CXXFLAGS "USE_IBVERBS=1")
 endif()
 mindspore_add_pkg(pslite
         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
         ONLY_MAKE True
         ONLY_MAKE_INCS include/*