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.

nccl.cmake 448 B

123456789101112
  1. set(nccl_CFLAGS "-D_FORTIFY_SOURCE=2 -O2")
  2. mindspore_add_pkg(nccl
  3. VER 2.4.8-1
  4. LIBS nccl
  5. URL https://github.com/NVIDIA/nccl/archive/v2.4.8-1.tar.gz
  6. MD5 f14b37d6af1c79db5f57cb029a753727
  7. BUILD_OPTION src.build NVCC_GENCODE="-gencode=arch=compute_70,code=sm_70"
  8. INSTALL_INCS build/include/*
  9. INSTALL_LIBS build/lib/*)
  10. include_directories(${nccl_INC})
  11. add_library(mindspore::nccl ALIAS nccl::nccl)

MindSpore is a new open source deep learning training/inference framework that could be used for mobile, edge and cloud scenarios.

Contributors (1)