From dd49368e0c617716d6993474f7fed95789ac35b5 Mon Sep 17 00:00:00 2001 From: Ruonan Wang Date: Thu, 5 Jun 2025 17:28:21 +0800 Subject: [PATCH] only install onednn for windows when torch 2.6 (#13207) --- python/llm/setup.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/python/llm/setup.py b/python/llm/setup.py index 11d47a20..6a03189e 100644 --- a/python/llm/setup.py +++ b/python/llm/setup.py @@ -304,10 +304,10 @@ def setup_package(): "torchvision==0.21.0+xpu", "torchaudio==2.6.0+xpu", "bigdl-core-xe-all==" + CORE_XE_VERSION, - "onednn-devel==2025.0.1", - "onednn==2025.0.1", + "onednn-devel==2025.0.1;platform_system=='Windows'", + "onednn==2025.0.1;platform_system=='Windows'", "dpcpp-cpp-rt==2025.0.2"] - + # Add for testing purposes for now, for Arrow Lake-H with AOT on Windows # Linux keeps the same as xpu_2.6 xpu_26_arl_requires = copy.deepcopy(all_requires) @@ -320,8 +320,8 @@ def setup_package(): "torchvision==0.21.0+xpu;platform_system=='Linux'", "torchaudio==2.6.0+xpu;platform_system=='Linux'", "bigdl-core-xe-all==" + CORE_XE_VERSION, - "onednn-devel==2025.0.1", - "onednn==2025.0.1", + "onednn-devel==2025.0.1;platform_system=='Windows'", + "onednn==2025.0.1;platform_system=='Windows'", "dpcpp-cpp-rt==2025.0.2"] cpp_requires = ["bigdl-core-cpp==" + CORE_XE_VERSION,