LLM: update setup.py to add bigdl-core-xe package (#9122)

intel-analytics · Oct 10, 2023 · 56f3fcd · 56f3fcd
1 parent f41ee66
commit 56f3fcd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/llm/setup.py b/python/llm/setup.py
@@ -261,7 +261,8 @@ def setup_package():
     xpu_requires += ["torch==2.0.1a0",
                      "torchvision==0.15.2a0",
                      "intel_extension_for_pytorch==2.0.110+xpu;platform_system=='Linux'",
-                     "bigdl-core-xe==" + VERSION + ";platform_system=='Linux'"]
+                     "bigdl-core-xe==" + VERSION + ";platform_system=='Linux'",
+                     "bigdl-core-xe-esimd==" + VERSION + ";platform_system=='Linux'"]
 
     serving_requires = ['py-cpuinfo']
     serving_requires += SERVING_DEP