Fix NVHPC builds.

This commit is contained in:
Ye Luo 2023-06-08 15:51:53 -05:00
parent 438e81a6d0
commit d624b52879
1 changed files with 4 additions and 2 deletions

View File

@ -36,11 +36,13 @@ endif()
if(QE_ENABLE_CUDA)
if(CMAKE_Fortran_COMPILER_VERSION VERSION_GREATER_EQUAL 20.7)
set(QE_CUDA_COMPILE_OPTIONS"-cuda")
set(CUDA_FLAG "-cuda")
else()
set(QE_CUDA_COMPILE_OPTIONS "-Mcuda")
set(CUDA_FLAG "-Mcuda")
endif()
set(QE_CUDA_COMPILE_OPTIONS ${CUDA_FLAG})
if(GPU_TARGET_COMPILE_OPTIONS)
list(APPEND QE_CUDA_COMPILE_OPTIONS ${GPU_TARGET_COMPILE_OPTIONS})
endif()