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

Fix cub

Browse Source
This commit is contained in:
Marcos Luciano
2022-06-21 12:40:45 -03:00
parent 0516a46138
commit de0c251b18
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
nvdsinfer_custom_impl_Yolo/sortDetections.cu
View File

@@ -3,7 +3,7 @@
* https://www.github.com/marcoslucianops
*/
#include <cub/cub.cuh>
#include <cub/device/device_radix_sort.cuh>
__global__ void sortOutput(
int* d_indexes, float* d_scores, float* d_boxes, int* d_classes, float* bboxData, float* scoreData,