add workflow

intel-analytics · May 27, 2024 · 29fc411 · 29fc411
1 parent 63cd866
commit 29fc411
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/.github/actions/llm/setup-llm-env/action.yml b/.github/actions/llm/setup-llm-env/action.yml
@@ -13,12 +13,20 @@ runs:
       run: |
         # make sure we install the latest version for bigdl-core-xe related packages
         pip uninstall bigdl-core-xe -y || true
+        pip uninstall bigdl-core-xe-batch -y || true
+        pip uninstall bigdl-core-xe-addons -y || true
         pip uninstall bigdl-core-xe-esimd -y || true
         pip uninstall bigdl-core-xe-21 -y || true
+        pip uninstall bigdl-core-xe-batch-21 -y || true
+        pip uninstall bigdl-core-xe-addons-21 -y || true
         pip uninstall bigdl-core-xe-esimd-21 -y || true
         sed -i 's/"bigdl-core-xe==" + CORE_XE_VERSION + "/"bigdl-core-xe/g' python/llm/setup.py
+        sed -i 's/"bigdl-core-xe-batch==" + CORE_XE_VERSION + "/"bigdl-core-xe-batch/g' python/llm/setup.py
+        sed -i 's/"bigdl-core-xe-addons==" + CORE_XE_VERSION + "/"bigdl-core-xe-addons/g' python/llm/setup.py
         sed -i 's/"bigdl-core-xe-esimd==" + CORE_XE_VERSION + "/"bigdl-core-xe-esimd/g' python/llm/setup.py
         sed -i 's/"bigdl-core-xe-21==" + CORE_XE_VERSION/"bigdl-core-xe-21"/g' python/llm/setup.py
+        sed -i 's/"bigdl-core-xe-batch-21==" + CORE_XE_VERSION/"bigdl-core-xe-batch-21"/g' python/llm/setup.py
+        sed -i 's/"bigdl-core-xe-addons-21==" + CORE_XE_VERSION/"bigdl-core-xe-addons-21"/g' python/llm/setup.py
         sed -i 's/"bigdl-core-xe-esimd-21==" + CORE_XE_VERSION/"bigdl-core-xe-esimd-21"/g' python/llm/setup.py
 
         pip install requests

diff --git a/.github/workflows/llm_performance_tests.yml b/.github/workflows/llm_performance_tests.yml
@@ -312,6 +312,8 @@ jobs:
       #   shell: bash
       #   run: |
       #     sed -i 's/"bigdl-core-xe-21==" + CORE_XE_VERSION/"bigdl-core-xe-21"/g' python/llm/setup.py
+            # sed -i 's/"bigdl-core-xe-batch-21==" + CORE_XE_VERSION/"bigdl-core-xe-batch-21"/g' python/llm/setup.py
+            # sed -i 's/"bigdl-core-xe-addons-21==" + CORE_XE_VERSION/"bigdl-core-xe-addons-21"/g' python/llm/setup.py
       #     sed -i 's/"bigdl-core-xe-esimd-21==" + CORE_XE_VERSION/"bigdl-core-xe-esimd-21"/g' python/llm/setup.py
 
       # - name: Install ipex-llm and other related packages (install from source)

diff --git a/python/llm/setup.py b/python/llm/setup.py
@@ -298,6 +298,8 @@ def setup_package():
                         "torchvision==0.16.0a0",
                         "intel_extension_for_pytorch==2.1.10+xpu",
                         "bigdl-core-xe-21==" + CORE_XE_VERSION,
+                        "bigdl-core-xe-batch-21==" + CORE_XE_VERSION,
+                        "bigdl-core-xe-addons-21==" + CORE_XE_VERSION,
                         "bigdl-core-xe-esimd-21==" + CORE_XE_VERSION]
     xpu_21_requires += oneapi_2024_0_requires
     # default to ipex 2.1 for linux and windows