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

Update setup.py for bigdl-core-xe-esimd-21 on Windows (#10705)

Browse source 
* Support bigdl-core-xe-esimd-21 for windows in setup.py

* Update setup-llm-env accordingly
This commit is contained in:
Yuwen Hu 2024-04-09 18:21:21 +08:00 committed by GitHub
parent b4147a97bb
commit 97db2492c8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -19,7 +19,7 @@ runs:
sed -i 's/"bigdl-core-xe==" + CORE_XE_VERSION + "/"bigdl-core-xe/g' python/llm/setup.py
sed -i 's/"bigdl-core-xe-esimd==" + CORE_XE_VERSION + "/"bigdl-core-xe-esimd/g' python/llm/setup.py
sed -i 's/"bigdl-core-xe-21==" + CORE_XE_VERSION/"bigdl-core-xe-21"/g' python/llm/setup.py
sed -i 's/"bigdl-core-xe-esimd-21==" + CORE_XE_VERSION + "/"bigdl-core-xe-esimd-21/g' python/llm/setup.py
sed -i 's/"bigdl-core-xe-esimd-21==" + CORE_XE_VERSION/"bigdl-core-xe-esimd-21"/g' python/llm/setup.py
pip install requests
if [[ ${{ runner.os }} == 'Linux' ]]; then

2
python/llm/setup.py
View file

@ -291,7 +291,7 @@ def setup_package():
"torchvision==0.16.0a0",
"intel_extension_for_pytorch==2.1.10+xpu",
"bigdl-core-xe-21==" + CORE_XE_VERSION,
"bigdl-core-xe-esimd-21==" + CORE_XE_VERSION + ";platform_system=='Linux'"]
"bigdl-core-xe-esimd-21==" + CORE_XE_VERSION]
# default to ipex 2.1 for linux and windows
xpu_requires = copy.deepcopy(xpu_21_requires)