From ba0483a633ff546e5abec18b821b4fc7143f7af8 Mon Sep 17 00:00:00 2001 From: Yuwen Hu Date: Thu, 16 Jan 2025 15:29:49 +0800 Subject: [PATCH] Update dependency for PyTorch 2.6 RC support for woq int4 --- python/llm/setup.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/llm/setup.py b/python/llm/setup.py index 92da2aa0ca5..d4c27a4bd32 100644 --- a/python/llm/setup.py +++ b/python/llm/setup.py @@ -320,7 +320,10 @@ def setup_package(): 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] + "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, "onednn-devel==2024.2.1;platform_system=='Windows'",