commit 1527d48cd0071e3e1737b51db3738f7f76ddbf80
author Robert Maynard <rmaynard@nvidia.com> Tue Jun 28 14:09:08 2022 -0400
committer Robert Maynard <rmaynard@nvidia.com> Wed Jul 13 10:25:10 2022 -0400
tree ad9dc16e53f06ded27c09adf5e17930777ae8f26
parent 4a552ab4ad181768fcbcb32a6c9ebf1383a8fe63

CheckIPO: Refactor logic selecting test source files

Modules/CheckIPOSupported.cmake

diff --git a/Modules/CheckIPOSupported.cmake b/Modules/CheckIPOSupported.cmake
index cca1be9..f0b4fd6 100644
--- a/Modules/CheckIPOSupported.cmake
+++ b/Modules/CheckIPOSupported.cmake
@@ -91,6 +91,17 @@
 
   set(try_compile_src "${CMAKE_ROOT}/Modules/CheckIPOSupported")
 
+  set(_C_ext "c")
+  set(_CXX_ext "cpp")
+  set(_Fortran_ext "f")
+
+  set(ext ${_${language}_ext})
+  if(NOT "${ext}" STREQUAL "")
+    set(copy_sources foo.${ext} main.${ext})
+  else()
+    message(FATAL_ERROR "Language not supported")
+  endif()
+
   # Use:
   # * TRY_COMPILE_PROJECT_NAME
   # * CMAKE_VERSION
@@ -100,20 +111,6 @@
       @ONLY
   )
 
-  string(COMPARE EQUAL "${language}" "C" is_c)
-  string(COMPARE EQUAL "${language}" "CXX" is_cxx)
-  string(COMPARE EQUAL "${language}" "Fortran" is_fortran)
-
-  if(is_c)
-    set(copy_sources foo.c main.c)
-  elseif(is_cxx)
-    set(copy_sources foo.cpp main.cpp)
-  elseif(is_fortran)
-    set(copy_sources foo.f main.f)
-  else()
-    message(FATAL_ERROR "Language not supported")
-  endif()
-
   foreach(x ${copy_sources})
     configure_file(
         "${try_compile_src}/${x}"