ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Make install ut for cpu exactly the same as what we want for users (#11051)

Browse source
This commit is contained in:
Yuwen Hu 2024-05-17 10:11:01 +08:00 committed by GitHub
parent 3a72e5df8c
commit fff067d240
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
.github/actions/llm/setup-llm-env/action.yml vendored
View file

@ -38,7 +38,11 @@ runs:
pip install --upgrade --pre -i https://pypi.python.org/simple --force-reinstall "python/llm/dist/${whl_name}[xpu_2.1]" --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/
pip install pytest expecttest
else
pip install --upgrade --pre -i https://pypi.python.org/simple --force-reinstall "python/llm/dist/${whl_name}[all]" --extra-index-url https://download.pytorch.org/whl/cpu
if [[ ${{ runner.os }} == 'Linux' ]]; then
pip install --upgrade --pre -i https://pypi.python.org/simple --force-reinstall "python/llm/dist/${whl_name}[all]" --extra-index-url https://download.pytorch.org/whl/cpu
elif [[ ${{ runner.os }} == 'Windows' ]]; then
pip install --upgrade --pre -i https://pypi.python.org/simple --force-reinstall "python/llm/dist/${whl_name}[all]"
fi
pip install pytest
bash python/llm/test/run-llm-install-tests.sh
fi