create macros.h for yolov5

Author: wang-xinyu

yolov5/API.h

@@ -11,7 +11,7 @@ set(CMAKE_BUILD_TYPE Debug)
 find_package(CUDA REQUIRED)
 
 if(WIN32)
-enable_language(CUDA) 
+enable_language(CUDA)
 endif(WIN32)
 
 include_directories(${PROJECT_SOURCE_DIR}/include)

yolov5/CMakeLists.txt

@@ -1,15 +1,10 @@
 #ifndef ENTROPY_CALIBRATOR_H
 #define ENTROPY_CALIBRATOR_H
 
-#include "NvInfer.h"
+#include <NvInfer.h>
 #include <string>
 #include <vector>
-
-#if NV_TENSORRT_MAJOR >= 8
-#define TRT_NOEXCEPT noexcept
-#else
-#define TRT_NOEXCEPT
-#endif
+#include "macros.h"
 
 //! \class Int8EntropyCalibrator2
 //!

yolov5/calibrator.h

@@ -25,12 +25,7 @@
 #include <ostream>
 #include <sstream>
 #include <string>
-
-#if NV_TENSORRT_MAJOR >= 8
-#define TRT_NOEXCEPT noexcept
-#else
-#define TRT_NOEXCEPT
-#endif
+#include "macros.h"
 
 using Severity = nvinfer1::ILogger::Severity;
 

yolov5/logging.h

@@ -0,0 +1,27 @@
+#ifndef __MACROS_H
+#define __MACROS_H
+
+#ifdef API_EXPORTS
+#if defined(_MSC_VER)
+#define API __declspec(dllexport)
+#else
+#define API __attribute__((visibility("default")))
+#endif
+#else
+
+#if defined(_MSC_VER)
+#define API __declspec(dllimport)
+#else
+#define API
+#endif
+#endif  // API_EXPORTS
+
+#if NV_TENSORRT_MAJOR >= 8
+#define TRT_NOEXCEPT noexcept
+#define TRT_CONST_ENQUEUE const
+#else
+#define TRT_NOEXCEPT
+#define TRT_CONST_ENQUEUE
+#endif
+
+#endif  // __MACROS_H

yolov5/macros.h

@@ -1,19 +1,10 @@
 #ifndef _YOLO_LAYER_H
 #define _YOLO_LAYER_H
 
-#include "API.h"
 #include <vector>
 #include <string>
-#include "NvInfer.h"
-
-
-#if NV_TENSORRT_MAJOR >= 8
-#define TRT_NOEXCEPT noexcept
-#define TRT_CONST_ENQUEUE const
-#else
-#define TRT_NOEXCEPT
-#define TRT_CONST_ENQUEUE
-#endif
+#include <NvInfer.h>
+#include "macros.h"
 
 namespace Yolo
 {
@@ -144,4 +135,4 @@ namespace nvinfer1
     REGISTER_TENSORRT_PLUGIN(YoloPluginCreator);
 };
 
-#endif 
+#endif  // _YOLO_LAYER_H