File tree Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Original file line number Diff line number Diff 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 :
Original file line number Diff line number Diff line change 44import logging
55import os
66import platform
7- import pwd
87import sys
98import tempfile
109from 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"
You can’t perform that action at this time.
0 commit comments