ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Support PyTorch 2.6 RC perf test on Windows (#12683)

Browse source
This commit is contained in:
Yuwen Hu 2025-01-09 18:17:23 +08:00 committed by GitHub
parent 7234c9b27b
commit c24741584d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 41 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
.github/workflows/llm_performance_tests.yml vendored
View file

@ -77,6 +77,10 @@ on:
required: false
type: boolean
default: true
gpu-pytorch-version:
description: 'PyTorch version used for GPU perf tests'
required: false
type: string
# A workflow run is made up of one or more jobs that can run sequentially or in parallel
jobs:
@ -649,6 +653,9 @@ jobs:
if: ${{ github.event_name == 'workflow_dispatch' && (inputs.checkout-ref != 'main') }}
shell: bash
run: |
if [ ${{ inputs.gpu-pytorch-version }} == "2.6" ]; then
sed -i 's/"bigdl-core-xe-all==" + CORE_XE_VERSION/"bigdl-core-xe-all"/g' python/llm/setup.py
else
if [ ${{ matrix.platform }} == "perf-mtl" ]; then
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
@ -659,6 +666,7 @@ jobs:
sed -i 's/"bigdl-core-xe-batch-23==" + CORE_XE_VERSION/"bigdl-core-xe-batch-23"/g' python/llm/setup.py
sed -i 's/"bigdl-core-xe-addons-23==" + CORE_XE_VERSION/"bigdl-core-xe-addons-23"/g' python/llm/setup.py
fi
fi
- name: Install ipex-llm and other related packages (install from source)
if: ${{ github.event_name == 'workflow_dispatch' && (inputs.checkout-ref != 'main') }}
@ -677,6 +685,9 @@ jobs:
if not exist dist\ipex_llm*.whl (exit /b 1)
for %%i in (dist\ipex_llm*.whl) do set whl_name=%%i
if "${{ inputs.gpu-pytorch-version }}"=="2.6" (
pip install --pre --upgrade %whl_name%[xpu_2.6] --extra-index-url https://download.pytorch.org/whl/test/xpu
) else (
if "${{ matrix.platform }}"=="perf-mtl" (
pip install --pre --upgrade %whl_name%[xpu] --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/
)
@ -686,6 +697,7 @@ jobs:
if "${{ matrix.platform }}"=="perf-dgpu" (
pip install --pre --upgrade %whl_name%[xpu_lnl] --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/lnl/cn/
)
)
if %ERRORLEVEL% neq 0 (exit /b 1)
pip list
@ -711,6 +723,9 @@ jobs:
pip install --upgrade omegaconf pandas
pip install --upgrade tiktoken einops transformers_stream_generator matplotlib
if "${{ inputs.gpu-pytorch-version }}"=="2.6" (
pip install --pre --upgrade ipex-llm[xpu_2.6] --extra-index-url https://download.pytorch.org/whl/test/xpu
) else (
if "${{ matrix.platform }}"=="perf-mtl" (
pip install --pre --upgrade ipex-llm[xpu]==%TEST_VERSION% --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/
)
@ -720,6 +735,7 @@ jobs:
if "${{ matrix.platform }}"=="perf-dgpu" (
pip install --pre --upgrade ipex-llm[xpu_lnl]==%TEST_VERSION% --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/lnl/cn/
)
)
pip show ipex-llm | findstr %TEST_VERSION%
if %ERRORLEVEL% neq 0 (
echo "Did not install ipex-llm with excepted version %TEST_VERSION%"