Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 7c3d64e0c9
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '7c3d64e0c9'
${ noResults }
mindspore/cmake/external_libs/nccl.cmake

19 lines
665 B
Raw Blame History 

  1. if(ENABLE_GITEE)

  2.     set(REQ_URL "https://gitee.com/mirrors/nccl/repository/archive/v2.7.6-1.tar.gz")

  3.     set(MD5 "220d232b30cb9bff2e54219399b9f6fb")

  4. else()

  5.     set(REQ_URL "https://github.com/NVIDIA/nccl/archive/v2.7.6-1.tar.gz")

  6.     set(MD5 "073b19899f374c5ba07d2db02dc38f9f")

  7. endif()

  8. 

  9. set(nccl_CFLAGS "-D_FORTIFY_SOURCE=2 -O2")

  10. mindspore_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/*)

  18. include_directories(${nccl_INC})

  19. add_library(mindspore::nccl ALIAS nccl::nccl)