Skip to content

Commit b165436

Browse files
authored
Merge branch 'develop' into gint-new-2
2 parents 2ffb5a2 + 37bc08a commit b165436

File tree

2 files changed

+12
-2
lines changed

2 files changed

+12
-2
lines changed

CMakeLists.txt

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff 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)

source/source_hsolver/kernels/cuda/diag_cusolver.cuh

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,14 @@
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

0 commit comments

Comments
 (0)