GeezerVison is showing github.com — try to link lib · TensorArray/TensorArray@f97f3f3
File tree
Expand file treeCollapse file tree
Open diff view settings
Expand file treeCollapse file tree
Open diff view settings
Collapse file
Copy file name to clipboardExpand all lines: cmake/ta_core_config.cmake +11-3Lines changed: 11 additions & 3 deletions | Original file line number | Diff line number | Diff line change |
|---|
@@ -28,6 +28,7 @@ if(CMAKE_CUDA_COMPILER) |
28 | 28 | set_property(TARGET tensorarray_core_object PROPERTY CUDA_STANDARD 17) |
29 | 29 | set_property(TARGET tensorarray_core_object PROPERTY CUDA_STANDARD_REQUIRED ON) |
30 | 30 | set_property(TARGET tensorarray_core_object PROPERTY CUDA_EXTENSIONS OFF) |
| 31 | + set_property(TARGET tensorarray_core_object PROPERTY CUDA_SEPARABLE_COMPILATION ON) |
31 | 32 | target_include_directories(tensorarray_core_object PRIVATE $<$<COMPILE_LANGUAGE:C,CXX>:${CUDAToolkit_INCLUDE_DIRS}>) |
32 | 33 | if(MSVC) |
33 | 34 | target_compile_definitions(tensorarray_core_object PRIVATE TENSOR_ARRAY_CORE_EXPORTS) |
@@ -59,9 +60,16 @@ add_library(tensorarray_core SHARED $<TARGET_OBJECTS:tensorarray_core_object>) |
59 | 60 | add_library(tensorarray_core_static STATIC $<TARGET_OBJECTS:tensorarray_core_object>) |
60 | 61 |
|
61 | 62 | if(CUDAToolkit_FOUND) |
62 | | - set_property(TARGET tensorarray_core PROPERTY CUDA_SEPARABLE_COMPILATION ON) |
63 | | - target_link_libraries(tensorarray_core PRIVATE $<$<LINK_LANGUAGE:C,CXX>:CUDA::cublas>) |
64 | | - target_link_libraries(tensorarray_core_static PRIVATE $<$<LINK_LANGUAGE:C,CXX>:CUDA::cublas>) |
| 63 | + target_link_libraries( |
| 64 | + tensorarray_core |
| 65 | + PRIVATE $<$<LINK_LANGUAGE:C,CXX>:CUDA::cudart> |
| 66 | + PRIVATE $<$<LINK_LANGUAGE:C,CXX>:CUDA::cublas> |
| 67 | + ) |
| 68 | + target_link_libraries( |
| 69 | + tensorarray_core_static |
| 70 | + PRIVATE $<$<LINK_LANGUAGE:C,CXX>:CUDA::cudart_static> |
| 71 | + PRIVATE $<$<LINK_LANGUAGE:C,CXX>:CUDA::cublas_static> |
| 72 | + ) |
65 | 73 | endif() |
66 | 74 |
|
67 | 75 | install( |
|