Include export header conditionally when compiling with CMake

Signed-off-by: Gabriel Cuendet <[email protected]>
Signed-off-by: Gabriel Cuendet <[email protected]>

Author: gcuendet-cognex

cpp/CMakeLists.txt

@@ -29,6 +29,8 @@ target_link_libraries(${lib_name}
         core
 )
 
+target_compile_definitions(${lib_name} PUBLIC "USE_CMAKE_GENERATED_EXPORT_HEADER")
+
 include("GenerateExportHeader")
 generate_export_header(${lib_name}
     EXPORT_MACRO_NAME TORCHTRT_API

cpp/include/torch_tensorrt/macros.h

@@ -7,14 +7,17 @@
  */
 #pragma once
 
+#if defined(USE_CMAKE_GENERATED_EXPORT_HEADER)
+#include <torch_tensorrt_export.h>
+#else
 #if defined(__GNUC__)
 #define TORCHTRT_API __attribute__((__visibility__("default")))
 #define TORCHTRT_HIDDEN __attribute__((__visibility__("hidden")))
 #else
-// This can be used also if defined(__GNUC__) but is generated by CMake
-// and will be missing when compiling with Bazel
-#include <torch_tensorrt_export.h>
+#define TORCHTRT_API
+#define TORCHTRT_HIDDEN
 #endif // defined(__GNUC__)
+#endif // defined(USE_CMAKE_GENERATED_EXPORT_HEADER)
 
 // Does this need to be gaurded or something?
 #define XSTR(x) #x