thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix cudaError_t error

Browse Source
This commit is contained in:
Marcos Luciano
2023-02-01 10:23:57 -03:00
parent 087a41acf6
commit 70c18a72a6
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
nvdsinfer_custom_impl_Yolo/yoloPlugins.h
View File

@@ -26,6 +26,8 @@
#ifndef __YOLO_PLUGINS__ #ifndef __YOLO_PLUGINS__
#define __YOLO_PLUGINS__ #define __YOLO_PLUGINS__
#include <cuda_runtime_api.h>
#include "yolo.h" #include "yolo.h"
#define CUDA_CHECK(status) { \ #define CUDA_CHECK(status) { \