Skip to content

Commit ab2e6e3

Browse files
move pwd under Linux branch (#3864)
1 parent c9859b6 commit ab2e6e3

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

.github/workflows/build-test-windows.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ jobs:
107107
set -euo pipefail
108108
pushd .
109109
cd tests/py/dynamo
110-
../../../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/
110+
../../../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/
111111
popd
112112
113113
L0-dynamo-core-tests:

py/torch_tensorrt/_TensorRTProxyModule.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44
import logging
55
import os
66
import platform
7-
import pwd
87
import sys
98
import tempfile
109
from types import ModuleType
@@ -74,6 +73,8 @@ def enable_capture_tensorrt_api_recording() -> None:
7473
os.environ["TRT_SHIM_NVINFER_LIB_NAME"] = os.path.join(
7574
tensorrt_lib_path, "libnvinfer.so"
7675
)
76+
import pwd
77+
7778
current_user = pwd.getpwuid(os.getuid())[0]
7879
shim_temp_dir = os.path.join(
7980
tempfile.gettempdir(), f"torch_tensorrt_{current_user}/shim"

0 commit comments

Comments
 (0)