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

Update dependency for PyTorch 2.6 RC support for woq int4 (#12714)

Browse source
This commit is contained in:
Yuwen Hu 2025-01-16 15:51:57 +08:00 committed by GitHub
parent 54d6328b3c
commit 534e0e6774
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
python/llm/setup.py
View file

@ -320,7 +320,10 @@ def setup_package():
xpu_26_requires += ["torch==2.6.0+xpu", xpu_26_requires += ["torch==2.6.0+xpu",
"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==2025.0.1",
"dpcpp-cpp-rt==2025.0.2"]
cpp_requires = ["bigdl-core-cpp==" + CORE_XE_VERSION, cpp_requires = ["bigdl-core-cpp==" + CORE_XE_VERSION,
"onednn-devel==2024.2.1;platform_system=='Windows'", "onednn-devel==2024.2.1;platform_system=='Windows'",