CMakeLists.txt 424 字节
Newer Older
Q
Qiao Longfei 已提交
1
cc_library(dynamic_loader SRCS dynamic_loader.cc DEPS glog gflags enforce)
2 3

list(APPEND CUDA_SRCS cublas.cc cudnn.cc curand.cc nccl.cc)
4
configure_file(cupti_lib_path.h.in ${CMAKE_CURRENT_BINARY_DIR}/cupti_lib_path.h)
5 6 7 8
if (CUPTI_FOUND)
    list(APPEND CUDA_SRCS cupti.cc)
endif(CUPTI_FOUND)
nv_library(dynload_cuda SRCS ${CUDA_SRCS} DEPS dynamic_loader)
Y
Yiqun Liu 已提交
9
cc_library(dynload_warpctc SRCS warpctc.cc DEPS dynamic_loader warpctc)