temp

MFlowCode · Mar 2, 2024 · a55a4e5 · a55a4e5
1 parent 928c5af
commit a55a4e5
Showing 1 changed file with 20 additions and 5 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -153,11 +153,20 @@ elseif ((CMAKE_Fortran_COMPILER_ID STREQUAL "NVHPC") OR (CMAKE_Fortran_COMPILER_
         add_compile_options(
             $<$<COMPILE_LANGUAGE:Fortran:-minline>
         )
+    elseif (CMAKE_BUILD_TYPE STREQUAL "Debug")
+        add_compile_options(
+            $<$<COMPILE_LANGUAGE:Fortran:-O0>
+        )
     endif()
-    
+
     if (CMAKE_BUILD_TYPE STREQUAL "Debug")
         add_compile_options(-C -g -O0 -traceback -Mchkptr -Mchkstk -Minform=inform -Mbounds)
     endif()
+
+    if (DEFINED ENV{MFC_CUDA_CC})
+        string(REGEX MATCHALL "[0-9]+" MFC_CUDA_CC $ENV{MFC_CUDA_CC})
+        message(STATUS "Found $MFC_CUDA_CC specified. GPU code will be generated for ${MFC_CUDA_CC}.")
+    endif()
 endif()
 
 if (CMAKE_BUILD_TYPE STREQUAL "Release")
@@ -395,13 +404,19 @@ function(MFC_SETUP_TARGET)
                 target_compile_definitions(${ARGS_TARGET} PRIVATE MFC_cuTENSOR)
             endif()
 
-            if     (CMAKE_BUILD_TYPE STREQUAL "Release")
+            foreach (cc ${MFC_CUDA_CC})
                 target_compile_options(${ARGS_TARGET}
-                    PRIVATE -gpu=keep,ptxinfo,lineinfo
+                    PRIVATE -gpu=cc${cc}
                 )
-            elseif (CMAKE_BUILD_TYPE STREQUAL "Debug")
+            endforeach()
+
+            target_compile_options(${ARGS_TARGET}
+                PRIVATE -gpu=keep,ptxinfo,lineinfo
+            )
+
+            if (CMAKE_BUILD_TYPE STREQUAL "Debug")
                 target_compile_options(${ARGS_TARGET}
-                    PRIVATE -gpu=keep,ptxinfo,lineinfo,autocompare,debug -O0
+                    PRIVATE -gpu=autocompare,debug
                 )
             endif()
         endif()