chore: import upstream snapshot with attribution

tools/Polygraphy/polygraphy/config.py

@@ -0,0 +1,65 @@
+#
+# 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.
+#
+import os
+import sys
+
+INTERNAL_CORRECTNESS_CHECKS = bool(
+    os.environ.get("POLYGRAPHY_INTERNAL_CORRECTNESS_CHECKS", "0") != "0"
+)
+"""
+bool: Whether internal correctness checks are enabled.
+This can be configured by setting the 'POLYGRAPHY_INTERNAL_CORRECTNESS_CHECKS' environment variable.
+"""
+
+AUTOINSTALL_DEPS = bool(os.environ.get("POLYGRAPHY_AUTOINSTALL_DEPS", "0") != "0")
+"""
+bool: Whether Polygraphy will automatically install required Python packages at runtime.
+This can be configured by setting the 'POLYGRAPHY_AUTOINSTALL_DEPS' environment variable.
+"""
+
+ASK_BEFORE_INSTALL = bool(os.environ.get("POLYGRAPHY_ASK_BEFORE_INSTALL", "0" != "0"))
+"""
+bool: Whether Polygraphy should ask before automatically installing required Python packages.
+Has no effect if AUTOINSTALL_DEPS is not enabled.
+This can be configured by setting the 'POLYGRAPHY_ASK_BEFORE_INSTALL' environment variable.
+"""
+
+INSTALL_CMD = os.environ.get(
+    "POLYGRAPHY_INSTALL_CMD", f"{sys.executable} -m pip install"
+).split()
+"""
+List[str]: The command to use to automatically install dependencies. Only relevant when
+AUTOINSTALL_DEPS is enabled. Defaults to ``["python", "-m", "pip", "install"]``.
+This can be configured by setting the 'POLYGRAPHY_INSTALL_CMD' environment variable to a
+string containing the command; for example: ``python3 -m pip install``.
+"""
+
+ARRAY_SWAP_THRESHOLD_MB = int(
+    os.environ.get("POLYGRAPHY_ARRAY_SWAP_THRESHOLD_MB", "-1")
+)
+"""
+int: The threshold, in megabytes, above which Polygraphy will evict an array from memory and swap it to disk.
+A negative value disables swapping and a value of 0 causes all arrays to be saved to disk.
+Disabled by default.
+This can be configured by setting the 'POLYGRAPHY_ARRAY_SWAP_THRESHOLD_MB' environment variable.
+"""
+
+USE_TENSORRT_RTX = bool(os.environ.get("POLYGRAPHY_USE_TENSORRT_RTX", "0") != "0")
+"""
+bool: Whether to use TensorRT RTX as the TensorRT backend.
+This can be configured by setting the 'POLYGRAPHY_USE_TENSORRT_RTX' environment variable.
+"""