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

only install onednn for windows when torch 2.6 (#13207)

Browse source
This commit is contained in:
Ruonan Wang 2025-06-05 17:28:21 +08:00 committed by GitHub
parent 5a1c1297e1
commit dd49368e0c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
python/llm/setup.py
View file

@ -304,10 +304,10 @@ def setup_package():
"torchvision==0.21.0+xpu", "torchvision==0.21.0+xpu",
"torchaudio==2.6.0+xpu", "torchaudio==2.6.0+xpu",
"bigdl-core-xe-all==" + CORE_XE_VERSION, "bigdl-core-xe-all==" + CORE_XE_VERSION,
"onednn-devel==2025.0.1", "onednn-devel==2025.0.1;platform_system=='Windows'",
"onednn==2025.0.1", "onednn==2025.0.1;platform_system=='Windows'",
"dpcpp-cpp-rt==2025.0.2"] "dpcpp-cpp-rt==2025.0.2"]
# Add for testing purposes for now, for Arrow Lake-H with AOT on Windows # Add for testing purposes for now, for Arrow Lake-H with AOT on Windows
# Linux keeps the same as xpu_2.6 # Linux keeps the same as xpu_2.6
xpu_26_arl_requires = copy.deepcopy(all_requires) xpu_26_arl_requires = copy.deepcopy(all_requires)
@ -320,8 +320,8 @@ def setup_package():
"torchvision==0.21.0+xpu;platform_system=='Linux'", "torchvision==0.21.0+xpu;platform_system=='Linux'",
"torchaudio==2.6.0+xpu;platform_system=='Linux'", "torchaudio==2.6.0+xpu;platform_system=='Linux'",
"bigdl-core-xe-all==" + CORE_XE_VERSION, "bigdl-core-xe-all==" + CORE_XE_VERSION,
"onednn-devel==2025.0.1", "onednn-devel==2025.0.1;platform_system=='Windows'",
"onednn==2025.0.1", "onednn==2025.0.1;platform_system=='Windows'",
"dpcpp-cpp-rt==2025.0.2"] "dpcpp-cpp-rt==2025.0.2"]
cpp_requires = ["bigdl-core-cpp==" + CORE_XE_VERSION, cpp_requires = ["bigdl-core-cpp==" + CORE_XE_VERSION,