File tree Expand file tree Collapse file tree 4 files changed +3
-4
lines changed Expand file tree Collapse file tree 4 files changed +3
-4
lines changed Original file line number Diff line number Diff line change @@ -72,6 +72,7 @@ FRAMEWORK="pytorch"
72
72
source /neural-compressor/.azure-pipelines/scripts/fwk_version.sh ' latest'
73
73
if [[ " ${inc_new_api} " == " 3x" * ]]; then
74
74
FRAMEWORK_VERSION=" latest"
75
+ export LD_LIBRARY_PATH=/usr/local/lib/:$LD_LIBRARY_PATH
75
76
else
76
77
FRAMEWORK_VERSION=${pytorch_version}
77
78
TORCH_VISION_VERSION=${torchvision_version}
Original file line number Diff line number Diff line change @@ -5,6 +5,7 @@ echo "${test_case}"
5
5
6
6
# install requirements
7
7
echo " set up UT env..."
8
+ export LD_LIBRARY_PATH=/usr/local/lib/:$LD_LIBRARY_PATH
8
9
pip install -r /neural-compressor/test/3x/torch/requirements.txt
9
10
pip install pytest-cov
10
11
pip install pytest-html
Original file line number Diff line number Diff line change @@ -6,6 +6,7 @@ echo "run itrex ut..."
6
6
7
7
# install inc 3x deps
8
8
pip install -r /neural-compressor/requirements_pt.txt
9
+ export LD_LIBRARY_PATH=/usr/local/lib/:$LD_LIBRARY_PATH
9
10
10
11
# prepare itrex
11
12
git clone https://github.com/intel/intel-extension-for-transformers.git /intel-extension-for-transformers
Original file line number Diff line number Diff line change 13
13
- requirements.txt
14
14
- .azure-pipelines/scripts/ut/run_itrex.sh
15
15
- .azure-pipelines/ut-itrex.yml
16
- exclude :
17
- - neural_compressor/common
18
- - neural_compressor/torch
19
- - neural_compressor/tensorflow
20
16
21
17
pool : MODEL_PERF_TEST
22
18
You can’t perform that action at this time.
0 commit comments