File tree Expand file tree Collapse file tree 2 files changed +12
-2
lines changed
source/source_hsolver/kernels/cuda Expand file tree Collapse file tree 2 files changed +12
-2
lines changed Original file line number Diff line number Diff line change @@ -351,7 +351,11 @@ if(USE_CUDA)
351351 # ${ABACUS_BIN_NAME} is added before CUDA is enabled
352352 set_property (TARGET ${ABACUS_BIN_NAME}
353353 PROPERTY CUDA_ARCHITECTURES ${CMAKE_CUDA_ARCHITECTURES} )
354- target_link_libraries (${ABACUS_BIN_NAME} cudart nvToolsExt)
354+ if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL 12.9)
355+ target_link_libraries (${ABACUS_BIN_NAME} cudart)
356+ else ()
357+ target_link_libraries (${ABACUS_BIN_NAME} cudart nvToolsExt)
358+ endif ()
355359 include_directories (${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES} )
356360 if (USE_CUDA)
357361 add_compile_definitions (__CUDA)
Original file line number Diff line number Diff line change 11#ifndef DIAG_CUSOLVER_CUH
22#define DIAG_CUSOLVER_CUH
3-
3+ # include < cuda.h >
44#include < complex>
5+
6+ #if CUDA_VERSION < 12090
57#include " nvToolsExt.h"
8+ #else
9+ #include " nvtx3/nvToolsExt.h"
10+ #endif
11+
612#include < cuda_runtime.h>
713#include < cusolverDn.h>
814
You can’t perform that action at this time.
0 commit comments