From 70c18a72a671a471b9ffe5bd926debe9faeb82c2 Mon Sep 17 00:00:00 2001
From: Marcos Luciano <marcoslucianops@gmail.com>
Date: Wed, 1 Feb 2023 10:23:57 -0300
Subject: [PATCH] Fix cudaError_t error

---
 nvdsinfer_custom_impl_Yolo/yoloPlugins.h | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/nvdsinfer_custom_impl_Yolo/yoloPlugins.h b/nvdsinfer_custom_impl_Yolo/yoloPlugins.h
index 7d41791..1293796 100644
--- 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) {                                                                                               \