update itrex ut test (#1929)

Signed-off-by: chensuyue <[email protected]>
intel · Jul 16, 2024 · f909bca · f909bca
1 parent 649e6b1
commit f909bca
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/.azure-pipelines/scripts/install_nc.sh b/.azure-pipelines/scripts/install_nc.sh
@@ -16,7 +16,7 @@ elif [[ $1 = *"3x_ort" ]]; then
     pip install dist/neural_compressor*.whl --force-reinstall
 else
     python -m pip install --no-cache-dir -r requirements.txt
-    python setup.py 2x bdist_wheel
+    python setup.py bdist_wheel
     pip install dist/neural_compressor*.whl --force-reinstall
 fi
 

diff --git a/.azure-pipelines/scripts/ut/run_itrex.sh b/.azure-pipelines/scripts/ut/run_itrex.sh
@@ -4,6 +4,9 @@ source /neural-compressor/.azure-pipelines/scripts/change_color.sh
 python -c "import neural_compressor as nc;print(nc.version.__version__)"
 echo "run itrex ut..."
 
+# install inc 3x deps
+pip install -r /neural-compressor/requirements_pt.txt
+
 # prepare itrex
 git clone https://github.com/intel/intel-extension-for-transformers.git /intel-extension-for-transformers
 cd /intel-extension-for-transformers && git rev-parse --short HEAD