chore: import upstream snapshot with attribution
Docker Image CI / build-ubuntu2004 (push) Waiting to run
Docker Image CI / build-ubuntu2004 (push) Waiting to run
This commit is contained in:
wehub-resource-sync
@@ -0,0 +1,168 @@
|
||||
# SPDX-FileCopyrightText: Copyright (c) 2025-2026 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.
|
||||
|
||||
include(FlagToInt)
|
||||
|
||||
# Processes one or more wheel templates file, replacing any markers with concrete information and
|
||||
# copying the result into the per-python per-module build dir.
|
||||
#
|
||||
# \param moduleName The module name to create the bindings for. One of "tensorrt", "tensorrt_dispatch", or "tensorrt_lean".
|
||||
# \param pyVersion The python version to create bindings for, i.e. "3.12".
|
||||
# \param ARGN A list of file paths, relative to './packaging/bindings_wheel/tensorrt', of the file(s) to copy.
|
||||
# \returns generatedWheelFiles A list containing paths to all generated files, which can be used to create a custom target.
|
||||
# \returns generatedFileOutDir A string containing the output directory for the generated files. Contains generator expressions.
|
||||
function(processWheelTemplates wheelType moduleName pyVersion)
|
||||
set(outputDir ${CMAKE_CURRENT_BINARY_DIR}/${moduleName}_${wheelType}-py${pyVersion}/$<CONFIG>)
|
||||
|
||||
foreach(filePath IN LISTS ARGN)
|
||||
# Target files that start with tensorrt/ need to be expanded into ${moduleName}/ instead of tensorrt/
|
||||
# So we need to rewrite the relative path, the source directory, *and* the output directory to match.
|
||||
# Note that we do not update the "outputDir" variable, since we want to report it back based on the original file path.
|
||||
if(filePath MATCHES "^tensorrt/")
|
||||
string(REGEX REPLACE "^tensorrt/" "" adjustedFilePath ${filePath})
|
||||
set(__srcDir ${CMAKE_CURRENT_LIST_DIR}/tensorrt)
|
||||
set(__srcFilePath ${adjustedFilePath})
|
||||
# Bit of a hack branch for standalone binding wheels
|
||||
if(wheelType STREQUAL "binding_standalone")
|
||||
set(__outDir ${outputDir}/${moduleName}_bindings)
|
||||
set(__outputFile ${outputDir}/${moduleName}_bindings/${adjustedFilePath})
|
||||
else()
|
||||
set(__outDir ${outputDir}/${moduleName})
|
||||
set(__outputFile ${outputDir}/${moduleName}/${adjustedFilePath})
|
||||
endif()
|
||||
elseif(filePath MATCHES "^tensorrt_libs/") # Standalone lib wheels use <module>_libs/ instead of <module>/
|
||||
string(REGEX REPLACE "^tensorrt_libs/" "" adjustedFilePath ${filePath})
|
||||
set(__srcDir ${CMAKE_CURRENT_LIST_DIR}/tensorrt_libs)
|
||||
set(__srcFilePath ${adjustedFilePath})
|
||||
set(__outDir ${outputDir}/${moduleName}_libs)
|
||||
set(__outputFile ${outputDir}/${moduleName}_libs/${adjustedFilePath})
|
||||
else()
|
||||
set(__srcDir ${CMAKE_CURRENT_LIST_DIR})
|
||||
set(__srcFilePath ${filePath})
|
||||
set(__outDir ${outputDir})
|
||||
set(__outputFile ${outputDir}/${filePath})
|
||||
endif()
|
||||
|
||||
flagToInt(TRT_BUILD_WINML)
|
||||
|
||||
add_custom_command(
|
||||
OUTPUT ${__outputFile}
|
||||
COMMAND
|
||||
${Python3_EXECUTABLE} ${TensorRT_SOURCE_DIR}/python/scripts/process_wheel_template.py
|
||||
--src-dir ${__srcDir}
|
||||
--dst-dir ${__outDir}
|
||||
--filepath ${__srcFilePath}
|
||||
--trt-module ${moduleName}
|
||||
--trt-py-version ${TensorRT_PACKAGE_VERSION}
|
||||
--cuda-version ${TRT_CUDA_VERSION}
|
||||
--trt-version ${TensorRT_VERSION}
|
||||
--plugin-disabled ${TRT_BUILD_WINML_INT}
|
||||
--trt-nvinfer-name ${TENSORRT_BASE_NAME}
|
||||
--trt-onnxparser-name ${TRT_ONNXPARSER_NAME} # The TRT_ONNXPARSER_NAME var is populated in the root CMakeLists.txt
|
||||
DEPENDS
|
||||
${TensorRT_SOURCE_DIR}/python/scripts/process_wheel_template.py
|
||||
${CMAKE_CURRENT_LIST_DIR}/${filePath}
|
||||
COMMENT "Expanding wheel template ${filePath} for ${moduleName}"
|
||||
VERBATIM
|
||||
)
|
||||
|
||||
list(APPEND generatedFiles ${__outputFile})
|
||||
endforeach()
|
||||
set(generatedWheelFiles ${generatedFiles} PARENT_SCOPE)
|
||||
set(generatedFileOutDir ${outputDir} PARENT_SCOPE)
|
||||
endfunction()
|
||||
|
||||
# Copy the custom build backend to the generated directory.
|
||||
# The backend is referenced via backend-path in pyproject.toml.
|
||||
# Note: This macro is intended to be called immediately after processWheelTemplates and relies on information generated by it.
|
||||
macro(copyTRTBuildBackend)
|
||||
set(copiedBuildBackend ${generatedFileOutDir}/tensorrt_build_backend/__init__.py)
|
||||
add_custom_command(
|
||||
OUTPUT ${copiedBuildBackend}
|
||||
COMMAND ${CMAKE_COMMAND} -E copy_directory
|
||||
${TensorRT_SOURCE_DIR}/python/packaging/tensorrt_build_backend
|
||||
${generatedFileOutDir}/tensorrt_build_backend
|
||||
DEPENDS ${TensorRT_SOURCE_DIR}/python/packaging/tensorrt_build_backend/__init__.py
|
||||
COMMENT "Copying tensorrt_build_backend to ${generatedFileOutDir}"
|
||||
VERBATIM
|
||||
)
|
||||
endmacro()
|
||||
|
||||
# Create meta targets used to group the post-processed wheel files and all the python wheels.
|
||||
add_custom_target(tensorrt_python_wheels)
|
||||
|
||||
# Specify a single output directory to hold all the generated wheels.
|
||||
# This is similar to CMAKE_ARCHIVE_OUTPUT_DIRECTORY, but for wheels.
|
||||
set(TRT_WHEEL_OUTPUT_DIR "${CMAKE_BINARY_DIR}/$<CONFIG>/dist" CACHE PATH "")
|
||||
|
||||
|
||||
# Install packaging requirements into the build dir so they survive container rebuilds.
|
||||
set(TRT_PACKAGING_DEPS_DIR ${CMAKE_CURRENT_BINARY_DIR}/packaging_deps)
|
||||
add_custom_command(
|
||||
OUTPUT ${TRT_PACKAGING_DEPS_DIR}/.installed
|
||||
COMMAND ${Python3_EXECUTABLE} -m pip install --upgrade
|
||||
--target ${TRT_PACKAGING_DEPS_DIR}
|
||||
-r ${TensorRT_SOURCE_DIR}/python/packaging/requirements.txt
|
||||
${maybe_pypi_index}
|
||||
COMMAND ${CMAKE_COMMAND} -E touch ${TRT_PACKAGING_DEPS_DIR}/.installed
|
||||
DEPENDS ${TensorRT_SOURCE_DIR}/python/packaging/requirements.txt
|
||||
VERBATIM
|
||||
)
|
||||
add_custom_target(trt_packaging_requirements_installed DEPENDS ${TRT_PACKAGING_DEPS_DIR}/.installed)
|
||||
|
||||
# Environment variables for Python packaging commands.
|
||||
set(TRT_PACKAGING_ENV "PYTHONPATH=${TRT_PACKAGING_DEPS_DIR}")
|
||||
if(NOT CMAKE_VERBOSE_MAKEFILE)
|
||||
# Suppress deprecation that we can't fix while python/packaging/requirements.txt
|
||||
# specifies setuptools~=75.3.2 for python_version<3.12.
|
||||
list(APPEND TRT_PACKAGING_ENV "PYTHONWARNINGS=ignore::setuptools.warnings.SetuptoolsDeprecationWarning")
|
||||
endif()
|
||||
|
||||
# Computes the current wheel platform name and stores it the variable named by outVar.
|
||||
#
|
||||
# \param isStandalone Boolean indicating whether to compute the platform for a standalone wheel.
|
||||
# \param outVar Name of the variable to store the resulting platform name in.
|
||||
# \returns wheelPlatform The computed wheel platform name. Standalone wheels use manylinux tags.
|
||||
function(get_wheel_platform isStandalone outVar)
|
||||
if(WIN32)
|
||||
set(wheelPlatform win_${TRT_LOWERCASE_CMAKE_SYSTEM_PROCESSOR})
|
||||
elseif(isStandalone)
|
||||
if(NOT GLIBC_VERSION)
|
||||
# Determine glibc version for standalone wheels
|
||||
execute_process(
|
||||
COMMAND ldd --version
|
||||
OUTPUT_VARIABLE LDD_OUTPUT
|
||||
OUTPUT_STRIP_TRAILING_WHITESPACE
|
||||
)
|
||||
|
||||
if(LDD_OUTPUT MATCHES "([0-9]+\\.[0-9]+)")
|
||||
set(GLIBC_VERSION ${CMAKE_MATCH_1} CACHE STRING "Detected glibc version for standalone wheels")
|
||||
else()
|
||||
message(FATAL_ERROR "Failed to determine glibc version from ldd output: ${LDD_OUTPUT}")
|
||||
endif()
|
||||
endif()
|
||||
|
||||
string(REPLACE "." "_" GLIBC_VERSION_FLAT ${GLIBC_VERSION})
|
||||
set(wheelPlatform manylinux_${GLIBC_VERSION_FLAT}_${TRT_LOWERCASE_CMAKE_SYSTEM_PROCESSOR})
|
||||
else()
|
||||
set(wheelPlatform linux_${TRT_LOWERCASE_CMAKE_SYSTEM_PROCESSOR})
|
||||
endif()
|
||||
|
||||
set(${outVar} ${wheelPlatform} PARENT_SCOPE)
|
||||
endfunction()
|
||||
|
||||
add_subdirectory(bindings_wheel)
|
||||
add_subdirectory(libs_wheel)
|
||||
add_subdirectory(frontend_sdist)
|
||||
add_subdirectory(metapackage)
|
||||
Reference in New Issue
Block a user