f2521a5ffa
* rcnn upgrade to support TensorRT 8. * Update RpnDecodePlugin.h Remove Chinese * Update rcnn.cpp Remove Chinese * Update backbone.hpp Remove Chinese * Update backbone.hpp * Update rcnn.cpp * Update rcnn.cpp * Update rcnn.cpp * Update MaskRcnnInferencePlugin.h * rcnn upgrade to support TensorRT 8.x * rcnn upgrade to support TensorRT 8.x * Update macros.h * Update README.md --------- Co-authored-by: nengwp <nengwp@github.ai> Co-authored-by: Wang Xinyu <shaywxy@gmail.com>
17 lines
252 B
C
17 lines
252 B
C
#pragma once
|
|
|
|
#include <NvInfer.h>
|
|
#include <cuda.h>
|
|
|
|
#if CUDA_VERSION >=11000
|
|
#define CUDA_11
|
|
#endif
|
|
|
|
#if NV_TENSORRT_MAJOR >= 8
|
|
#define TRT_NOEXCEPT noexcept
|
|
#define TRT_CONST_ENQUEUE const
|
|
#else
|
|
#define TRT_NOEXCEPT
|
|
#define TRT_CONST_ENQUEUE
|
|
#endif
|