Browse Source

!30593 correct cublas path in cuda_ops cmakelists

Merge pull request !30593 from jinjiali-kali/cuda_ops
feature/build-system-rewrite
i-robot Gitee 4 years ago
parent
commit
09f114e52e
No known key found for this signature in database GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 3 deletions
  1. +2
    -3
      mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/CMakeLists.txt

+ 2
- 3
mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/CMakeLists.txt View File

@@ -16,12 +16,11 @@ if(ENABLE_GPU)
target_compile_options(cuda_common_obj PRIVATE "-std=c++17")
cuda_add_library(cuda_ops SHARED ${CUDA_OPS_SRC_LIST} $<TARGET_OBJECTS:cuda_common_obj>)
message("add gpu lib to cuda_ops")
target_link_libraries(cuda_ops mindspore_core
target_link_libraries(cuda_ops mindspore_core cublas
${CUDA_PATH}/lib64/libcurand.so
${CUDNN_LIBRARY_PATH}
${CUDA_PATH}/lib64/libcudart.so
${CUDA_PATH}/lib64/stubs/libcuda.so
${CUDA_PATH}/lib64/libcusolver.so
${CUDA_PATH}/lib64/libcufft.so
${CUDA_PATH}/lib64/libcublas.so)
${CUDA_PATH}/lib64/libcufft.so)
endif()

Loading…
Cancel
Save