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: 930a1fb0a8
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '930a1fb0a8'
${ noResults }
mindspore/cmake/external_libs/nccl.cmake

12 lines
448 B
Raw Blame History 

  1. 

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

  3. mindspore_add_pkg(nccl

  4.         VER 2.4.8-1

  5.         LIBS nccl

  6.         URL https://github.com/NVIDIA/nccl/archive/v2.4.8-1.tar.gz

  7.         MD5 f14b37d6af1c79db5f57cb029a753727

  8.         BUILD_OPTION src.build NVCC_GENCODE="-gencode=arch=compute_70,code=sm_70"

  9.         INSTALL_INCS build/include/*

  10.         INSTALL_LIBS build/lib/*)

  11. include_directories(${nccl_INC})

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