From f9ee7898c87cc533a654733f77068c7f32a83157 Mon Sep 17 00:00:00 2001 From: Yishuo Wang Date: Tue, 7 Jan 2025 16:26:56 +0800 Subject: [PATCH] fix onednn dependency bug (#12665) --- python/llm/setup.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/python/llm/setup.py b/python/llm/setup.py index f135f8cf..c2554bd0 100644 --- a/python/llm/setup.py +++ b/python/llm/setup.py @@ -310,15 +310,18 @@ def setup_package(): "bigdl-core-xe-23==" + CORE_XE_VERSION, "bigdl-core-xe-batch-23==" + CORE_XE_VERSION, "bigdl-core-xe-addons-23==" + CORE_XE_VERSION, - "onednn-devel==2024.1.1;platform_system=='Windows'"] + "onednn-devel==2024.1.1;platform_system=='Windows'", + "onednn==2024.1.1;platform_system=='Windows'"] cpp_requires = ["bigdl-core-cpp==" + CORE_XE_VERSION, - "onednn-devel==2024.2.1;platform_system=='Windows'"] + "onednn-devel==2024.2.1;platform_system=='Windows'", + "onednn==2024.2.1;platform_system=='Windows'"] cpp_requires += oneapi_2024_2_requires cpp_requires += COMMON_DEP cpp_arl_requires = ["bigdl-core-cpp==" + CORE_XE_VERSION, - "onednn-devel==2024.1.1;platform_system=='Windows'"] + "onednn-devel==2024.1.1;platform_system=='Windows'", + "onednn==2024.1.1;platform_system=='Windows'"] cpp_arl_requires += oneapi_2024_2_requires cpp_arl_requires += COMMON_DEP