chore: import upstream snapshot with attribution

2026-07-13 13:36:55 +08:00
commit c8a779b1bb
1887 changed files with 3245738 additions and 0 deletions
@@ -0,0 +1,50 @@
+#!/usr/bin/env bash
+#
+# 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.
+#
+
+arg_dockerfile=docker/ubuntu-24.04.Dockerfile
+arg_imagename=tensorrt-ubuntu
+arg_help=0
+
+while [[ "$#" -gt 0 ]]; do case $1 in
+  --file) arg_dockerfile="$2"; shift;;
+  --tag) arg_imagename="$2"; shift;;
+  --cuda) arg_cudaversion="$2"; shift;;
+  -h|--help) arg_help=1;;
+  *) echo "Unknown parameter passed: $1"; echo "For help type: $0 --help"; exit 1;
+esac; shift; done
+
+if [ "$arg_help" -eq "1" ]; then
+    echo "Usage: $0 [options]"
+    echo " --help or -h         : Print this help menu."
+    echo " --file  <dockerfile> : Docker file to use for build."
+    echo " --tag   <imagename>  : Image name for the generated container."
+    echo " --cuda  <version>    : CUDA version to use."
+    exit;
+fi
+
+if [ -z "$arg_cudaversion" ]
+then
+    echo "--cuda not specified, so not passing in --build-arg CUDA_VERSION to Dockerfile"
+    docker_args="-f $arg_dockerfile --build-arg uid=$(id -u) --build-arg gid=$(id -g) --tag=$arg_imagename ."
+else
+    docker_args="-f $arg_dockerfile --build-arg CUDA_VERSION=$arg_cudaversion --build-arg CUDA_VERSION_MAJOR_MINOR=${arg_cudaversion:0:4} --build-arg uid=$(id -u) --build-arg gid=$(id -g) --tag=$arg_imagename ."
+fi
+
+echo "Building container:"
+echo "> docker build $docker_args"
+docker build $docker_args