mirror of
https://github.com/Kitware/CMake.git
synced 2026-05-09 07:39:47 -05:00
FindCUDA: Add cublas device library to separable compilation
Add cublas libraries to separable compilation device link line to resolve missing symbols when using cublas device library.
This commit is contained in:
@@ -1756,7 +1756,7 @@ function(CUDA_LINK_SEPARABLE_COMPILATION_OBJECTS output_file cuda_target options
|
|||||||
add_custom_command(
|
add_custom_command(
|
||||||
OUTPUT ${output_file}
|
OUTPUT ${output_file}
|
||||||
DEPENDS ${object_files}
|
DEPENDS ${object_files}
|
||||||
COMMAND ${CUDA_NVCC_EXECUTABLE} ${nvcc_flags} -dlink ${object_files} -o ${output_file}
|
COMMAND ${CUDA_NVCC_EXECUTABLE} ${nvcc_flags} -dlink ${object_files} ${CUDA_cublas_device_LIBRARY} -o ${output_file}
|
||||||
${flags}
|
${flags}
|
||||||
COMMENT "Building NVCC intermediate link file ${output_file_relative_path}"
|
COMMENT "Building NVCC intermediate link file ${output_file_relative_path}"
|
||||||
COMMAND_EXPAND_LISTS
|
COMMAND_EXPAND_LISTS
|
||||||
@@ -1769,7 +1769,7 @@ function(CUDA_LINK_SEPARABLE_COMPILATION_OBJECTS output_file cuda_target options
|
|||||||
PRE_LINK
|
PRE_LINK
|
||||||
COMMAND ${CMAKE_COMMAND} -E echo "Building NVCC intermediate link file ${output_file_relative_path}"
|
COMMAND ${CMAKE_COMMAND} -E echo "Building NVCC intermediate link file ${output_file_relative_path}"
|
||||||
COMMAND ${CMAKE_COMMAND} -E make_directory "${output_file_dir}"
|
COMMAND ${CMAKE_COMMAND} -E make_directory "${output_file_dir}"
|
||||||
COMMAND ${CUDA_NVCC_EXECUTABLE} ${nvcc_flags} ${flags} -dlink ${object_files} -o "${output_file}"
|
COMMAND ${CUDA_NVCC_EXECUTABLE} ${nvcc_flags} ${flags} -dlink ${object_files} ${CUDA_cublas_device_LIBRARY} -o "${output_file}"
|
||||||
COMMAND_EXPAND_LISTS
|
COMMAND_EXPAND_LISTS
|
||||||
${_verbatim}
|
${_verbatim}
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user