From cd3ab90867c19d8c73d83a3a3933a6457b834177 Mon Sep 17 00:00:00 2001
From: Marcos Luciano <marcoslucianops@gmail.com>
Date: Thu, 2 Feb 2023 09:56:33 -0300
Subject: [PATCH] Fix cudaError_t error

---
 nvdsinfer_custom_impl_Yolo/calibrator.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/nvdsinfer_custom_impl_Yolo/calibrator.h b/nvdsinfer_custom_impl_Yolo/calibrator.h
index 2a18f83..e390a76 100644
--- a/nvdsinfer_custom_impl_Yolo/calibrator.h
+++ b/nvdsinfer_custom_impl_Yolo/calibrator.h
@@ -7,6 +7,7 @@
 #define CALIBRATOR_H
 
 #include <vector>
+#include <cuda_runtime_api.h>
 
 #include "NvInfer.h"
 #include "opencv2/opencv.hpp"