From 97db2492c82900e9b2e73c88fe6db4109c3eeb56 Mon Sep 17 00:00:00 2001 From: Yuwen Hu <54161268+Oscilloscope98@users.noreply.github.com> Date: Tue, 9 Apr 2024 18:21:21 +0800 Subject: [PATCH] Update setup.py for `bigdl-core-xe-esimd-21` on Windows (#10705) * Support bigdl-core-xe-esimd-21 for windows in setup.py * Update setup-llm-env accordingly --- .github/actions/llm/setup-llm-env/action.yml | 2 +- python/llm/setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/actions/llm/setup-llm-env/action.yml b/.github/actions/llm/setup-llm-env/action.yml index 1d530972..4b25ea0c 100644 --- a/.github/actions/llm/setup-llm-env/action.yml +++ b/.github/actions/llm/setup-llm-env/action.yml @@ -19,7 +19,7 @@ runs: sed -i 's/"bigdl-core-xe==" + CORE_XE_VERSION + "/"bigdl-core-xe/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-esimd-21==" + CORE_XE_VERSION + "/"bigdl-core-xe-esimd-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 if [[ ${{ runner.os }} == 'Linux' ]]; then diff --git a/python/llm/setup.py b/python/llm/setup.py index 02fabbbb..48689570 100644 --- a/python/llm/setup.py +++ b/python/llm/setup.py @@ -291,7 +291,7 @@ 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-esimd-21==" + CORE_XE_VERSION + ";platform_system=='Linux'"] + "bigdl-core-xe-esimd-21==" + CORE_XE_VERSION] # default to ipex 2.1 for linux and windows xpu_requires = copy.deepcopy(xpu_21_requires)