diff --git a/.github/workflows/build-test-windows.yml b/.github/workflows/build-test-windows.yml index 86324f68f9..874d472fe2 100644 --- a/.github/workflows/build-test-windows.yml +++ b/.github/workflows/build-test-windows.yml @@ -107,7 +107,7 @@ jobs: set -euo pipefail pushd . cd tests/py/dynamo - ../../../packaging/vc_env_helper.bat python -m pytest -ra -n 8--junitxml=${RUNNER_TEST_RESULTS_DIR}/l0_dynamo_converter_tests_results.xml --dist=loadscope --maxfail=20 conversion/ + ../../../packaging/vc_env_helper.bat python -m pytest -ra -n 8 --junitxml=${RUNNER_TEST_RESULTS_DIR}/l0_dynamo_converter_tests_results.xml --dist=loadscope --maxfail=20 conversion/ popd L0-dynamo-core-tests: diff --git a/py/torch_tensorrt/_TensorRTProxyModule.py b/py/torch_tensorrt/_TensorRTProxyModule.py index 84d2c50f5f..ff7077e05e 100644 --- a/py/torch_tensorrt/_TensorRTProxyModule.py +++ b/py/torch_tensorrt/_TensorRTProxyModule.py @@ -4,7 +4,6 @@ import logging import os import platform -import pwd import sys import tempfile from types import ModuleType @@ -74,6 +73,8 @@ def enable_capture_tensorrt_api_recording() -> None: os.environ["TRT_SHIM_NVINFER_LIB_NAME"] = os.path.join( tensorrt_lib_path, "libnvinfer.so" ) + import pwd + current_user = pwd.getpwuid(os.getuid())[0] shim_temp_dir = os.path.join( tempfile.gettempdir(), f"torch_tensorrt_{current_user}/shim"