forked from mindspore-Ecosystem/mindspore
12 lines
448 B
CMake
12 lines
448 B
CMake
|
|
||
|
|
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
|
||
|
|
BUILD_OPTION src.build NVCC_GENCODE="-gencode=arch=compute_70,code=sm_70"
|
||
|
|
INSTALL_INCS build/include/*
|
||
|
|
INSTALL_LIBS build/lib/*)
|
||
|
|
include_directories(${nccl_INC})
|
||
|
|
add_library(mindspore::nccl ALIAS nccl::nccl)
|