From 8d8b65ea50a613ddf51f6d6849dd8557f4859f3b Mon Sep 17 00:00:00 2001 From: Yuwen Hu Date: Mon, 28 Apr 2025 13:40:08 +0800 Subject: [PATCH 1/2] Add support of PyTorch 2.7 for XPU --- python/llm/setup.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/python/llm/setup.py b/python/llm/setup.py index 11d47a2044a..7c151bd0f83 100644 --- a/python/llm/setup.py +++ b/python/llm/setup.py @@ -323,6 +323,16 @@ def setup_package(): "onednn-devel==2025.0.1", "onednn==2025.0.1", "dpcpp-cpp-rt==2025.0.2"] + + xpu_27_requires = copy.deepcopy(all_requires) + for exclude_require in cpu_torch_version: + xpu_26_requires.remove(exclude_require) + xpu_26_requires += ["torch==2.7.0+xpu", + "torchvision==0.22.0+xpu", + "torchaudio==2.7.0+xpu", + "bigdl-core-xe-all==" + CORE_XE_VERSION, + "onednn-devel==2025.1.0", + "onednn==2025.1.0"] cpp_requires = ["bigdl-core-cpp==" + CORE_XE_VERSION, "onednn-devel==2025.0.1;platform_system=='Windows'", @@ -368,6 +378,7 @@ def setup_package(): "xpu-2-1": xpu_21_requires, "xpu-2-6": xpu_26_requires, "xpu-2-6-arl": xpu_26_arl_requires, + "xpu-2-7": xpu_27_requires, "serving": serving_requires, "cpp": cpp_requires, "llama-index": llama_index_requires}, # for internal usage when upstreaming for llama-index From e0fa858353953a93c339f82984f1cd88e3f9e319 Mon Sep 17 00:00:00 2001 From: Yuwen Hu Date: Mon, 28 Apr 2025 14:04:52 +0800 Subject: [PATCH 2/2] Small fix --- python/llm/setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/llm/setup.py b/python/llm/setup.py index 7c151bd0f83..9fa3cb1cc71 100644 --- a/python/llm/setup.py +++ b/python/llm/setup.py @@ -326,8 +326,8 @@ def setup_package(): xpu_27_requires = copy.deepcopy(all_requires) for exclude_require in cpu_torch_version: - xpu_26_requires.remove(exclude_require) - xpu_26_requires += ["torch==2.7.0+xpu", + xpu_27_requires.remove(exclude_require) + xpu_27_requires += ["torch==2.7.0+xpu", "torchvision==0.22.0+xpu", "torchaudio==2.7.0+xpu", "bigdl-core-xe-all==" + CORE_XE_VERSION,