fix onednn dependency bug (#12665)

intel-analytics · Jan 7, 2025 · f9ee789 · f9ee789
1 parent 29ad5c4
commit f9ee789
Showing 1 changed file with 6 additions and 3 deletions.
diff --git 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