Skip to content

Commit b0329ec

Browse files
authored
Add files via upload
1 parent 673cbb4 commit b0329ec

File tree

1 file changed

+60
-0
lines changed
  • DeepStream/nvdsinfer_custom_impl_Yolo

1 file changed

+60
-0
lines changed
Lines changed: 60 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,60 @@
1+
################################################################################
2+
# Copyright (c) 2019, NVIDIA CORPORATION. All rights reserved.
3+
#
4+
# Permission is hereby granted, free of charge, to any person obtaining a
5+
# copy of this software and associated documentation files (the "Software"),
6+
# to deal in the Software without restriction, including without limitation
7+
# the rights to use, copy, modify, merge, publish, distribute, sublicense,
8+
# and/or sell copies of the Software, and to permit persons to whom the
9+
# Software is furnished to do so, subject to the following conditions:
10+
#
11+
# The above copyright notice and this permission notice shall be included in
12+
# all copies or substantial portions of the Software.
13+
#
14+
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15+
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16+
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
17+
# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18+
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
19+
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
20+
# DEALINGS IN THE SOFTWARE.
21+
################################################################################
22+
23+
CUDA_VER?=
24+
ifeq ($(CUDA_VER),)
25+
$(error "CUDA_VER is not set")
26+
endif
27+
CC:= g++
28+
NVCC:=/usr/local/cuda-$(CUDA_VER)/bin/nvcc
29+
30+
CFLAGS:= -Wall -std=c++11 -shared -fPIC -Wno-error=deprecated-declarations
31+
CFLAGS+= -I../../includes -I/usr/local/cuda-$(CUDA_VER)/include
32+
33+
LIBS:= -lnvinfer_plugin -lnvinfer -lnvparsers -L/usr/local/cuda-$(CUDA_VER)/lib64 -lcudart -lcublas -lstdc++fs
34+
LFLAGS:= -shared -Wl,--start-group $(LIBS) -Wl,--end-group
35+
36+
INCS:= $(wildcard *.h)
37+
SRCFILES:= nvdsinfer_yolo_engine.cpp \
38+
nvdsparsebbox_Yolo.cpp \
39+
yoloPlugins.cpp \
40+
trt_utils.cpp \
41+
yolo.cpp \
42+
kernels.cu
43+
TARGET_LIB:= libnvdsinfer_custom_impl_Yolo.so
44+
45+
TARGET_OBJS:= $(SRCFILES:.cpp=.o)
46+
TARGET_OBJS:= $(TARGET_OBJS:.cu=.o)
47+
48+
all: $(TARGET_LIB)
49+
50+
%.o: %.cpp $(INCS) Makefile
51+
$(CC) -c -o $@ $(CFLAGS) $<
52+
53+
%.o: %.cu $(INCS) Makefile
54+
$(NVCC) -c -o $@ --compiler-options '-fPIC' $<
55+
56+
$(TARGET_LIB) : $(TARGET_OBJS)
57+
$(CC) -o $@ $(TARGET_OBJS) $(LFLAGS)
58+
59+
clean:
60+
rm -rf $(TARGET_LIB)

0 commit comments

Comments
 (0)