42b4f94527
* yolov3 trt8 fix * move the if,else to macro header like yolov5 * reset imagesize to 608 and other stuff
13 lines
211 B
C
13 lines
211 B
C
#ifndef __MACROS_H
|
|
#define __MACROS_H
|
|
|
|
#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
|