export LD_LIBRARY_PATH in test

Signed-off-by: chensuyue <[email protected]>
intel · Jul 29, 2024 · 6103791 · 6103791
1 parent 5226448
commit 6103791
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/.azure-pipelines/scripts/models/run_pytorch_models_trigger.sh b/.azure-pipelines/scripts/models/run_pytorch_models_trigger.sh
@@ -72,6 +72,7 @@ FRAMEWORK="pytorch"
 source /neural-compressor/.azure-pipelines/scripts/fwk_version.sh 'latest'
 if [[ "${inc_new_api}" == "3x"* ]]; then
     FRAMEWORK_VERSION="latest"
+    export LD_LIBRARY_PATH=/usr/local/lib/:$LD_LIBRARY_PATH
 else
     FRAMEWORK_VERSION=${pytorch_version}
     TORCH_VISION_VERSION=${torchvision_version}

diff --git a/.azure-pipelines/scripts/ut/3x/run_3x_pt.sh b/.azure-pipelines/scripts/ut/3x/run_3x_pt.sh
@@ -5,6 +5,7 @@ echo "${test_case}"
 
 # install requirements
 echo "set up UT env..."
+export LD_LIBRARY_PATH=/usr/local/lib/:$LD_LIBRARY_PATH
 pip install -r /neural-compressor/test/3x/torch/requirements.txt
 pip install pytest-cov
 pip install pytest-html

diff --git a/.azure-pipelines/ut-itrex.yml b/.azure-pipelines/ut-itrex.yml
@@ -13,10 +13,6 @@ pr:
       - requirements.txt
       - .azure-pipelines/scripts/ut/run_itrex.sh
       - .azure-pipelines/ut-itrex.yml
-    exclude:
-      - neural_compressor/common
-      - neural_compressor/torch
-      - neural_compressor/tensorflow
 
 pool: MODEL_PERF_TEST