diff --git a/python/llm/setup.py b/python/llm/setup.py
index be80ea23..8f2a3cd4 100644
--- a/python/llm/setup.py
+++ b/python/llm/setup.py
@@ -253,7 +253,8 @@ def setup_package():
     xpu_requires.remove('torch')
     xpu_requires += ["torch==2.0.1a0",
                      "torchvision==0.15.2a0",
-                     "intel_extension_for_pytorch==2.0.110+xpu;platform_system=='Linux'"]
+                     "intel_extension_for_pytorch==2.0.110+xpu;platform_system=='Linux'",
+                     "bigdl-core-xe;platform_system=='Linux'"]
 
     metadata = dict(
         name='bigdl-llm',