From 56f3fcd6f7a6189837042be02006873d6896d15d Mon Sep 17 00:00:00 2001
From: Ruonan Wang <105281011+rnwang04@users.noreply.github.com>
Date: Tue, 10 Oct 2023 15:02:48 +0800
Subject: [PATCH] LLM: update setup.py to add `bigdl-core-xe` package (#9122)

---
 python/llm/setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/llm/setup.py b/python/llm/setup.py
index ebdc4b664ea..a90aa4daa4c 100644
--- 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