30 lines
484 B
C
30 lines
484 B
C
#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
|