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

6
nvdsinfer_custom_impl_Yolo/Makefile
View File

@@ -83,8 +83,12 @@ all: $(TARGET_LIB)
%.o: %.cpp $(INCS) Makefile
$(CC) -c $(COMMON) -o $@ $(CFLAGS) $<
ifeq ($(CUDA_VER), 10.2)
CUB=-I/usr/local/cuda-$(CUDA_VER)/include/thrust/system/cuda/detail
endif
%.o: %.cu $(INCS) Makefile
$(NVCC) -c -o $@ --compiler-options '-fPIC' $<
$(NVCC) -c -o $@ $(CUB) --compiler-options '-fPIC' $<
$(TARGET_LIB) : $(TARGET_OBJS)
$(CC) -o $@ $(TARGET_OBJS) $(LFLAGS)

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,