1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
| #ifndef __MACROS_H
| #define __MACROS_H
|
| #include "NvInfer.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
|
|
