e73bffcd25
* tensorrt-yolov9 * format change * change format * Update block.cpp * format: add space * update block.cpp: add space * update config.h --------- Co-authored-by: Wang Xinyu <wangxinyu_es@163.com>
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
|