From 5db6f9dcde122c51730ba4ac96ef7df347ff35f8 Mon Sep 17 00:00:00 2001 From: Yuwen Hu <54161268+Oscilloscope98@users.noreply.github.com> Date: Tue, 7 Jan 2025 18:28:55 +0800 Subject: [PATCH] Add option with PyTorch 2.6 RC version for testing purposes (#12668) * Add option with PyTorch 2.6 RC version for testing purposes * Small update --- python/llm/setup.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/python/llm/setup.py b/python/llm/setup.py index c2554bd0062..92da2aa0ca5 100644 --- a/python/llm/setup.py +++ b/python/llm/setup.py @@ -312,6 +312,15 @@ def setup_package(): "bigdl-core-xe-addons-23==" + CORE_XE_VERSION, "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'", @@ -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,