chore: import upstream snapshot with attribution
Docker Image CI / build-ubuntu2004 (push) Has been cancelled
Docker Image CI / build-ubuntu2004 (push) Has been cancelled
This commit is contained in:
wehub-resource-sync
@@ -0,0 +1,242 @@
|
||||
#
|
||||
# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
|
||||
.SUFFIXES:
|
||||
CUDA_TRIPLE ?= x86_64-linux
|
||||
CUBLAS_TRIPLE ?= x86_64-linux-gnu
|
||||
DLSW_TRIPLE ?= x86_64-linux-gnu
|
||||
TARGET ?= $(shell uname -m)
|
||||
|
||||
ifeq ($(CUDA_INSTALL_DIR),)
|
||||
CUDA_INSTALL_DIR ?= /usr/local/cuda
|
||||
$(warning CUDA_INSTALL_DIR variable is not specified, using $(CUDA_INSTALL_DIR) by default, use CUDA_INSTALL_DIR=<cuda_directory> to change.)
|
||||
endif
|
||||
|
||||
ifeq ($(CUDNN_INSTALL_DIR),)
|
||||
CUDNN_INSTALL_DIR ?= $(CUDA_INSTALL_DIR)
|
||||
$(warning CUDNN_INSTALL_DIR variable is not specified, using $(CUDA_INSTALL_DIR) by default, use CUDNN_INSTALL_DIR=<cudnn_directory> to change.)
|
||||
endif
|
||||
|
||||
|
||||
ifeq ($(TRT_LIB_DIR),)
|
||||
TRT_LIB_DIR ?= /usr/lib/x86_64-linux-gnu/
|
||||
$(warning TRT_LIB_DIR is not specified, searching $(TRT_LIB_DIR), /usr/lib/x86_64-linux-gnu/ by default, use TRT_LIB_DIR=<trt_lib_directory> to change.)
|
||||
endif
|
||||
|
||||
ifeq ($(TRT_INC_DIR),)
|
||||
TRT_INC_DIR ?= /usr/include/x86_64-linux-gnu/
|
||||
$(warning TRT_INC_DIR is not specified, searching $(TRT_INC_DIR), /usr/include/x86_64-linux-gnu/ by default, use TRT_INC_DIR=<trt_include_directory> to change.)
|
||||
endif
|
||||
|
||||
CUDA_LIBDIR = lib
|
||||
CUDNN_LIBDIR = lib64
|
||||
ifeq ($(TARGET), aarch64)
|
||||
ifeq ($(shell uname -m), aarch64)
|
||||
CUDA_LIBDIR = lib64
|
||||
CC = g++
|
||||
else
|
||||
CC = aarch64-linux-gnu-g++
|
||||
endif
|
||||
CUCC = $(CUDA_INSTALL_DIR)/bin/nvcc -m64 -ccbin $(CC)
|
||||
else ifeq ($(TARGET), x86_64)
|
||||
CUDA_LIBDIR = lib64
|
||||
CC = g++
|
||||
CUCC = $(CUDA_INSTALL_DIR)/bin/nvcc -m64
|
||||
else ifeq ($(TARGET), ppc64le)
|
||||
CUDA_LIBDIR = lib64
|
||||
CC = g++
|
||||
CUCC = $(CUDA_INSTALL_DIR)/bin/nvcc -m64
|
||||
else ifeq ($(TARGET), qnx)
|
||||
CC = ${QNX_HOST}/usr/bin/aarch64-unknown-nto-qnx7.0.0-g++
|
||||
CUCC = $(CUDA_INSTALL_DIR)/bin/nvcc -m64 -ccbin $(CC)
|
||||
else ifeq ($(TARGET), android64)
|
||||
ifeq ($(ANDROID_CC),)
|
||||
$(error ANDROID_CC must be set to the clang compiler to build for android 64bit, for example /path/to/my-toolchain/bin/aarch64-linux-android-clang++)
|
||||
endif
|
||||
CUDA_LIBDIR = lib
|
||||
ANDROID_FLAGS = -DANDROID -D_GLIBCXX_USE_C99=1 -Wno-sign-compare -D__aarch64__ -Wno-strict-aliasing -Werror -pie -fPIE -Wno-unused-command-line-argument
|
||||
COMMON_FLAGS += $(ANDROID_FLAGS)
|
||||
COMMON_LD_FLAGS += $(ANDROID_FLAGS)
|
||||
CC = $(ANDROID_CC)
|
||||
CUCC = $(CUDA_INSTALL_DIR)/bin/nvcc -m64 -ccbin $(CC) --compiler-options="-DANDROID -D_GLIBCXX_USE_C99=1 -Wno-sign-compare"
|
||||
ANDROID = 1
|
||||
else ########
|
||||
$(error Auto-detection of platform failed. Please specify one of the following arguments to make: TARGET=[aarch64|x86_64|qnx|android64])
|
||||
endif
|
||||
|
||||
ifdef VERBOSE
|
||||
AT=
|
||||
else
|
||||
AT=@
|
||||
endif
|
||||
|
||||
AR = ar cr
|
||||
ECHO = @echo
|
||||
|
||||
SHELL = /bin/sh
|
||||
|
||||
ROOT_PATH = .
|
||||
OUTDIR = $(ROOT_PATH)/bin
|
||||
|
||||
define concat
|
||||
$1$2$3$4$5$6$7$8
|
||||
endef
|
||||
|
||||
ifneq ($(USE_QCC),1)
|
||||
# Usage: $(call make-depend,source-file,object-file,depend-file)
|
||||
define make-depend
|
||||
$(AT)$(CC) -MM -MF $3 -MP -MT $2 $(COMMON_FLAGS) $1
|
||||
endef
|
||||
# Usage: $(call make-cuda-depend,source-file,object-file,depend-file,flags)
|
||||
define make-cuda-depend
|
||||
$(AT)$(CUCC) -M -MT $2 $4 $1 > $3
|
||||
endef
|
||||
endif
|
||||
|
||||
CUDART_LIB = -lcudart
|
||||
CUDNN_LIB = -lcudnn
|
||||
CUBLAS_LIB = -lcublas
|
||||
NVINFER_LIB = -lnvinfer
|
||||
NVINFER_PLUGIN_LIB = -lnvinfer_plugin
|
||||
NVONNXPARSERS_LIB = -lnvonnxparser
|
||||
NVRTC_LIB = -lnvrtc
|
||||
PROTO_LIBDIR =
|
||||
STUBS_DIR =
|
||||
|
||||
#########################
|
||||
INCPATHS=
|
||||
LIBPATHS=
|
||||
COMMON_LIBS=
|
||||
|
||||
# Add extra libraries if TRT_STATIC is enabled
|
||||
ifeq ($(TRT_STATIC), 1)
|
||||
COMMON_LIBS += -lculibos -lcublasLt_static
|
||||
endif
|
||||
|
||||
# add cross compile directories
|
||||
ifneq ($(shell uname -m), $(TARGET))
|
||||
INCPATHS += -I"/usr/include/$(DLSW_TRIPLE)" -I"/usr/include/$(CUBLAS_TRIPLE)"
|
||||
LIBPATHS += -L"../lib/stubs" -L"../../lib/stubs" -L"/usr/lib/$(DLSW_TRIPLE)/stubs" -L"/usr/lib/$(DLSW_TRIPLE)" -L"/usr/lib/$(CUBLAS_TRIPLE)/stubs" -L"/usr/lib/$(CUBLAS_TRIPLE)"
|
||||
LIBPATHS += -L"$(CUDA_INSTALL_DIR)/targets/$(CUDA_TRIPLE)/$(CUDA_LIBDIR)/stubs" -L"$(CUDA_INSTALL_DIR)/targets/$(CUDA_TRIPLE)/$(CUDA_LIBDIR)"
|
||||
endif
|
||||
INCPATHS += -I"../common" -I"$(TRT_INC_DIR)" -I"$(CUDA_INSTALL_DIR)/include" -I"$(CUDNN_INSTALL_DIR)/include"
|
||||
LIBPATHS += -L"$(CUDA_INSTALL_DIR)/$(CUDA_LIBDIR)" -Wl,-rpath-link="$(CUDA_INSTALL_DIR)/$(CUDA_LIBDIR)"
|
||||
LIBPATHS += -L"$(CUDNN_INSTALL_DIR)/$(CUDNN_LIBDIR)" -Wl,-rpath-link="$(CUDNN_INSTALL_DIR)/$(CUDNN_LIBDIR)"
|
||||
LIBPATHS += -L"$(TRT_LIB_DIR)" -Wl,-rpath-link="$(TRT_LIB_DIR)" $(STUBS_DIR)
|
||||
|
||||
.SUFFIXES:
|
||||
vpath %.h $(EXTRA_DIRECTORIES)
|
||||
vpath %.cpp $(EXTRA_DIRECTORIES)
|
||||
|
||||
COMMON_FLAGS += -Wall -Wno-deprecated-declarations -std=c++11 $(INCPATHS)
|
||||
ifneq ($(ANDROID),1)
|
||||
COMMON_FLAGS += -D_REENTRANT
|
||||
endif
|
||||
|
||||
ifeq ($(TARGET), qnx)
|
||||
COMMON_FLAGS += -D_POSIX_C_SOURCE=200112L -D_QNX_SOURCE -D_FILE_OFFSET_BITS=64 -fpermissive
|
||||
endif
|
||||
|
||||
COMMON_LD_FLAGS += $(LIBPATHS) -L$(OUTDIR)
|
||||
|
||||
OBJDIR = $(call concat,$(OUTDIR),/chobj)
|
||||
DOBJDIR = $(call concat,$(OUTDIR),/dchobj)
|
||||
|
||||
COMMON_LIBS += $(CUDART_LIB) $(CUBLAS_LIB) $(CUDNN_LIB)
|
||||
|
||||
ifneq ($(TARGET), qnx)
|
||||
ifneq ($(ANDROID), 1)
|
||||
COMMON_LIBS += -lrt -ldl -lpthread
|
||||
endif
|
||||
endif
|
||||
ifeq ($(ANDROID),1)
|
||||
COMMON_LIBS += -lculibos -llog
|
||||
endif
|
||||
|
||||
COMMON_LIBS_FOR_EXECUTABLE := $(filter-out -lcudart_static ,$(COMMON_LIBS))
|
||||
ifeq ($(USE_CUDART_STATIC), 1)
|
||||
COMMON_LIBS_FOR_EXECUTABLE += $(CUDART_LIB)
|
||||
endif
|
||||
|
||||
LIBS = $(NVINFER_LIB) $(NVINFER_PLUGIN_LIB) $(NVONNXPARSERS_LIB) $(COMMON_LIBS_FOR_EXECUTABLE) $(PROTO_LIB) $(EXTRA_LIBS)
|
||||
DLIBS = $(NVINFER_LIB) $(NVINFER_PLUGIN_LIB) $(NVONNXPARSERS_LIB) $(COMMON_LIBS_FOR_EXECUTABLE) $(PROTO_LIB) $(EXTRA_LIBS)
|
||||
|
||||
OBJS = $(patsubst %.cpp, $(OBJDIR)/%.o, $(wildcard *.cpp $(addsuffix /*.cpp, $(EXTRA_DIRECTORIES))))
|
||||
DOBJS = $(patsubst %.cpp, $(DOBJDIR)/%.o, $(wildcard *.cpp $(addsuffix /*.cpp, $(EXTRA_DIRECTORIES))))
|
||||
CUOBJS = $(patsubst %.cu, $(OBJDIR)/%.o, $(wildcard *.cu $(addsuffix /*.cu, $(EXTRA_DIRECTORIES))))
|
||||
CUDOBJS = $(patsubst %.cu, $(DOBJDIR)/%.o, $(wildcard *.cu $(addsuffix /*.cu, $(EXTRA_DIRECTORIES))))
|
||||
|
||||
CFLAGS = $(COMMON_FLAGS)
|
||||
CFLAGSD = $(COMMON_FLAGS) -g
|
||||
LFLAGS = $(COMMON_LD_FLAGS)
|
||||
LFLAGSD = $(COMMON_LD_FLAGS)
|
||||
|
||||
all: debug release
|
||||
|
||||
release : $(OUTDIR)/$(OUTNAME_RELEASE)
|
||||
|
||||
debug : $(OUTDIR)/$(OUTNAME_DEBUG)
|
||||
|
||||
$(OUTDIR)/$(OUTNAME_RELEASE) : $(OBJS) $(CUOBJS)
|
||||
$(ECHO) Linking: $@
|
||||
$(AT)$(CC) -o $@ $^ $(LFLAGS) -Wl,--start-group $(LIBS) -Wl,--end-group
|
||||
# Copy every EXTRA_FILE of this sample to bin dir
|
||||
$(foreach EXTRA_FILE,$(EXTRA_FILES), cp -f $(EXTRA_FILE) $(OUTDIR)/$(EXTRA_FILE); )
|
||||
|
||||
$(OUTDIR)/$(OUTNAME_DEBUG) : $(DOBJS) $(CUDOBJS)
|
||||
$(ECHO) Linking: $@
|
||||
$(AT)$(CC) -o $@ $^ $(LFLAGSD) -Wl,--start-group $(DLIBS) -Wl,--end-group
|
||||
|
||||
$(OBJDIR)/%.o: %.cpp
|
||||
$(AT)if [ ! -d $(OBJDIR) ]; then mkdir -p $(OBJDIR); fi
|
||||
$(foreach XDIR,$(EXTRA_DIRECTORIES), if [ ! -d $(OBJDIR)/$(XDIR) ]; then mkdir -p $(OBJDIR)/$(XDIR); fi;) :
|
||||
$(call make-depend,$<,$@,$(subst .o,.d,$@))
|
||||
$(ECHO) Compiling: $<
|
||||
$(AT)$(CC) $(CFLAGS) -c -o $@ $<
|
||||
|
||||
$(DOBJDIR)/%.o: %.cpp
|
||||
$(AT)if [ ! -d $(DOBJDIR) ]; then mkdir -p $(DOBJDIR); fi
|
||||
$(foreach XDIR,$(EXTRA_DIRECTORIES), if [ ! -d $(OBJDIR)/$(XDIR) ]; then mkdir -p $(DOBJDIR)/$(XDIR); fi;) :
|
||||
$(call make-depend,$<,$@,$(subst .o,.d,$@))
|
||||
$(ECHO) Compiling: $<
|
||||
$(AT)$(CC) $(CFLAGSD) -c -o $@ $<
|
||||
|
||||
######################################################################### CU
|
||||
$(OBJDIR)/%.o: %.cu
|
||||
$(AT)if [ ! -d $(OBJDIR) ]; then mkdir -p $(OBJDIR); fi
|
||||
$(foreach XDIR,$(EXTRA_DIRECTORIES), if [ ! -d $(OBJDIR)/$(XDIR) ]; then mkdir -p $(OBJDIR)/$(XDIR); fi;) :
|
||||
$(call make-cuda-depend,$<,$@,$(subst .o,.d,$@))
|
||||
$(ECHO) Compiling CUDA release: $<
|
||||
$(AT)$(CUCC) $(CUFLAGS) -c -o $@ $<
|
||||
|
||||
$(DOBJDIR)/%.o: %.cu
|
||||
$(AT)if [ ! -d $(DOBJDIR) ]; then mkdir -p $(DOBJDIR); fi
|
||||
$(foreach XDIR,$(EXTRA_DIRECTORIES), if [ ! -d $(DOBJDIR)/$(XDIR) ]; then mkdir -p $(DOBJDIR)/$(XDIR); fi;) :
|
||||
$(call make-cuda-depend,$<,$@,$(subst .o,.d,$@))
|
||||
$(ECHO) Compiling CUDA debug: $<
|
||||
$(AT)$(CUCC) $(CUFLAGSD) -c -o $@ $<
|
||||
|
||||
clean:
|
||||
$(ECHO) Cleaning...
|
||||
$(foreach XDIR,$(EXTRA_DIRECTORIES), if [ -d $(OBJDIR)/$(XDIR) ]; then rm -rf $(OBJDIR)/$(XDIR); fi;) :
|
||||
$(foreach XDIR,$(EXTRA_DIRECTORIES), if [ -d $(DOBJDIR)/$(XDIR) ]; then rm -rf $(DOBJDIR)/$(XDIR); fi;) :
|
||||
$(AT)rm -rf $(OBJDIR) $(DOBJDIR) $(OUTDIR)/$(OUTNAME_RELEASE) $(OUTDIR)/$(OUTNAME_DEBUG)
|
||||
$(foreach EXTRA_FILE,$(EXTRA_FILES), if [ -f $(OUTDIR)/$(EXTRA_FILE) ]; then rm -f $(OUTDIR)/$(EXTRA_FILE); fi;) :
|
||||
|
||||
ifneq "$(MAKECMDGOALS)" "clean"
|
||||
-include $(OBJDIR)/*.d $(DOBJDIR)/*.d
|
||||
endif # ifneq "$(MAKECMDGOALS)" "clean"
|
||||
Reference in New Issue
Block a user