Added CMake config for NVIDIA HPC compilers (nvfortran)

Reference-LAPACK · Aug 24, 2023 · 64d95ed · 64d95ed
1 parent 3f1aef1
commit 64d95ed
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/CMAKE/CheckLAPACKCompilerFlags.cmake b/CMAKE/CheckLAPACKCompilerFlags.cmake
@@ -31,6 +31,12 @@ elseif( CMAKE_Fortran_COMPILER_ID STREQUAL "NAG" )
     else ()
         set(FOPT_ILP64 -i8)
     endif()
+elseif( CMAKE_Fortran_COMPILER_ID STREQUAL "NVHPC" )
+    if ( WIN32 )
+        set(FOPT_ILP64 /i8)
+    else ()
+        set(FOPT_ILP64 -i8)
+    endif()
 else()
     set(CPE_ENV $ENV{PE_ENV})
     if(CPE_ENV STREQUAL "CRAY")
@@ -138,6 +144,17 @@ elseif( CMAKE_Fortran_COMPILER_ID STREQUAL "NAG" )
     set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -quiet")
   endif()
 
+# NVIDIA HPC SDK
+elseif( CMAKE_Fortran_COMPILER_ID STREQUAL "NVHPC" )
+  if( ("${CMAKE_Fortran_FLAGS}" MATCHES "-Ktrap=") AND
+      NOT ("${CMAKE_Fortran_FLAGS}" MATCHES "-Ktrap=none") )
+    set( FPE_EXIT TRUE )
+  endif()
+
+  if( NOT ("${CMAKE_Fortran_FLAGS}" MATCHES "[-/]Kieee") )
+    set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -Kieee")
+  endif()
+
 else()
 endif()
 

diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -141,6 +141,8 @@ elseif(CMAKE_Fortran_COMPILER_ID STREQUAL XL)
   check_fortran_compiler_flag("-qrecur" _qrecurFlag)
 elseif(CMAKE_Fortran_COMPILER_ID STREQUAL NAG)
   check_fortran_compiler_flag("-recursive" _recursiveFlag)
+elseif(CMAKE_Fortran_COMPILER_ID STREQUAL NVHPC)
+  check_fortran_compiler_flag("-Mrecursive" _MrecursiveFlag)
 else()
   message(WARNING "Fortran local arrays should be allocated on the stack."
     " Please use a compiler which guarantees that feature."