Add build configured torchfort_config.h header to indicate GPU support in build.

CMakeLists.txt

@@ -114,6 +114,13 @@ endif()
 
 find_package(Torch REQUIRED)
 
+# Generate configuration header
+configure_file(
+  ${CMAKE_CURRENT_SOURCE_DIR}/src/csrc/include/torchfort_config.h.in
+  ${CMAKE_BINARY_DIR}/include/torchfort_config.h
+  @ONLY
+)
+
 # yaml-cpp
 #find_package(yaml-cpp REQUIRED)
 find_path(YAML_CPP_INCLUDE_DIR REQUIRED
@@ -170,6 +177,7 @@ target_include_directories(${PROJECT_NAME}
   ${CMAKE_CURRENT_SOURCE_DIR}/src/csrc/include
   PUBLIC
   $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/src/csrc/include>
+  $<BUILD_INTERFACE:${CMAKE_BINARY_DIR}/include>
   $<INSTALL_INTERFACE:${CMAKE_INSTALL_PREFIX}/include>
 )
 target_link_libraries(${PROJECT_NAME} PRIVATE ${TORCH_LIBRARIES})
@@ -209,6 +217,12 @@ install(
   INCLUDES DESTINATION ${CMAKE_INSTALL_PREFIX}/include
 )
 
+# Install generated configuration header
+install(
+  FILES ${CMAKE_BINARY_DIR}/include/torchfort_config.h
+  DESTINATION ${CMAKE_INSTALL_PREFIX}/include
+)
+
 # Fortran library and module
 if (TORCHFORT_BUILD_FORTRAN)
 

src/csrc/include/torchfort.h

@@ -16,8 +16,9 @@
  */
 
 #pragma once
+#include "torchfort_config.h"
 #include <stdint.h>
-#ifdef ENABLE_GPU
+#if TORCHFORT_ENABLE_GPU
 #include <cuda_runtime.h>
 #else
 typedef void* cudaStream_t;

src/csrc/include/torchfort_config.h.in

@@ -0,0 +1,22 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2023-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#pragma once
+
+/* Define to 1 if TorchFort was built with GPU support */
+#cmakedefine01 TORCHFORT_ENABLE_GPU
+

src/csrc/include/torchfort_rl.h

@@ -16,8 +16,9 @@
  */
 
 #pragma once
+#include "torchfort_config.h"
 #include "torchfort_enums.h"
-#ifdef ENABLE_GPU
+#if TORCHFORT_ENABLE_GPU
 #include <cuda_runtime.h>
 #else
 typedef void* cudaStream_t;