e4da5a8367
* detr添加对trt8的支持 * 修改格式 * 修改格式 * 修改格式 * 修改格式 * Update detr.cpp --------- Co-authored-by: Wang Xinyu <shaywxy@gmail.com>
30 lines
485 B
C
30 lines
485 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)a
|
|
#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
|