duan8/yolov3-tiny/macros.h
Dominic D 46dccf11ea
Update YOLOv3-tiny for TRT8 (#1064)
* Update YOLOv3-tiny for TRT8
Changes from #967

* Also apply fixes to utils

* Remove redundant macro definition in logging.g
2022-08-02 13:33:13 +08:00

12 lines
210 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