TensorRT-Demo/plugins/Makefile

38 lines
1.1 KiB
Makefile
Raw Permalink Normal View History

2023-03-06 20:44:29 +08:00
CC=g++
LD=ld
CXXFLAGS=-Wall -std=c++11 -g -O
NVCC=nvcc
# space separated compute values ex: computes=70 75. If not present will fetch device's CC
computes=
ifeq ($(computes), )
computes= $(shell python gpu_cc.py)
$(info computes: $(computes))
endif
NVCCFLAGS= $(foreach compute, $(computes),-gencode arch=compute_$(compute),code=[sm_$(compute),compute_$(compute)])
$(info NVCCFLAGS: $(NVCCFLAGS))
# These are the directories where I installed TensorRT on my x86_64 PC.
TENSORRT_INCS=-I"/usr/local/TensorRT-7.1.3.4/include"
TENSORRT_LIBS=-L"/usr/local/TensorRT-7.1.3.4/lib"
# INCS and LIBS
INCS=-I"/usr/local/cuda/include" $(TENSORRT_INCS) -I"/usr/local/include" -I"plugin"
LIBS=-L"/usr/local/cuda/lib64" $(TENSORRT_LIBS) -L"/usr/local/lib" -Wl,--start-group -lnvinfer -lnvparsers -lnvinfer_plugin -lcudnn -lcublas -lnvToolsExt -lcudart -lrt -ldl -lpthread -Wl,--end-group
.PHONY: all clean
all: libyolo_layer.so
clean:
rm -f *.so *.o
libyolo_layer.so: yolo_layer.o
$(CC) -shared -o $@ $< $(LIBS)
yolo_layer.o: yolo_layer.cu yolo_layer.h
$(NVCC) -ccbin $(CC) $(INCS) $(NVCCFLAGS) -Xcompiler -fPIC -c -o $@ $<