Use the export macros generated by CMake on Windows

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

Author: gcuendet-cognex

cpp/include/torch_tensorrt/macros.h

@@ -11,8 +11,9 @@
 #define TORCHTRT_API __attribute__((__visibility__("default")))
 #define TORCHTRT_HIDDEN __attribute__((__visibility__("hidden")))
 #else
-#define TORCHTRT_API
-#define TORCHTRT_HIDDEN
+// This can be used also if defined(__GNUC__) but is generated by CMake
+// and will be missing when compiling with Bazel
+#include <torchtrt_export.h>
 #endif // defined(__GNUC__)
 
 // Does this need to be gaurded or something?
@@ -31,4 +32,4 @@ namespace torch_tensorrt {
 namespace torchscript {}
 namespace ts = torchscript;
 } // namespace torch_tensorrt
-namespace torchtrt = torch_tensorrt;
+namespace torchtrt = torch_tensorrt;