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

Add option with PyTorch 2.6 RC version for testing purposes (#12668)

Browse source 
* Add option with PyTorch 2.6 RC version for testing purposes

* Small update
This commit is contained in:
Yuwen Hu 2025-01-07 18:28:55 +08:00 committed by GitHub
parent f9ee7898c8
commit 5db6f9dcde
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
python/llm/setup.py
View file

@ -313,6 +313,15 @@ def setup_package():
"onednn-devel==2024.1.1;platform_system=='Windows'",
"onednn==2024.1.1;platform_system=='Windows'"]
# Add for testing purposes for now
xpu_26_requires = copy.deepcopy(all_requires)
for exclude_require in cpu_torch_version:
xpu_26_requires.remove(exclude_require)
xpu_26_requires += ["torch==2.6.0+xpu",
"torchvision==0.21.0+xpu",
"torchaudio==2.6.0+xpu",
"bigdl-core-xe-all==" + CORE_XE_VERSION]
cpp_requires = ["bigdl-core-cpp==" + CORE_XE_VERSION,
"onednn-devel==2024.2.1;platform_system=='Windows'",
"onednn==2024.2.1;platform_system=='Windows'"]
@ -363,6 +372,7 @@ def setup_package():
"xpu-lnl": xpu_lnl_requires,
"xpu-arl": xpu_lnl_requires,
"xpu-arc": xpu_lnl_requires,
"xpu-2-6": xpu_26_requires,
"serving": serving_requires,
"cpp": cpp_requires,
"cpp-arl": cpp_arl_requires,