• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1if(ENABLE_GITEE)
2    set(REQ_URL "https://gitee.com/mirrors/nccl/repository/archive/v2.7.6-1.tar.gz")
3    set(MD5 "0f2797ab8c876840d09d1277bdead339")
4else()
5    set(REQ_URL "https://github.com/NVIDIA/nccl/archive/v2.7.6-1.tar.gz")
6    set(MD5 "073b19899f374c5ba07d2db02dc38f9f")
7endif()
8
9set(nccl_CFLAGS "-D_FORTIFY_SOURCE=2 -O2")
10mindspore_add_pkg(nccl
11        VER 2.7.6-1
12        LIBS nccl
13        URL ${REQ_URL}
14        MD5 ${MD5}
15        BUILD_OPTION src.build NVCC_GENCODE="-gencode=arch=compute_70,code=sm_70"
16        INSTALL_INCS build/include/*
17        INSTALL_LIBS build/lib/*)
18include_directories(${nccl_INC})
19add_library(mindspore::nccl ALIAS nccl::nccl)