Fix cudaError_t error

2023-02-01 10:23:57 -03:00
parent 087a41acf6
commit 70c18a72a6
1 changed files with 2 additions and 0 deletions
--- a/nvdsinfer_custom_impl_Yolo/yoloPlugins.h
+++ b/nvdsinfer_custom_impl_Yolo/yoloPlugins.h
@@ -26,6 +26,8 @@
 #ifndef __YOLO_PLUGINS__
 #define __YOLO_PLUGINS__

+#include <cuda_runtime_api.h>
+
 #include "yolo.h"

 #define CUDA_CHECK(status) {                                                                                               \