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

Enable llm gpu tests for PyTorch 2.1 (#9863)

Browse source
This commit is contained in:
Yuwen Hu 2024-01-09 16:29:02 +08:00 committed by GitHub
parent 23fc888abe
commit aebed4b7bc
1 changed files with 12 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
.github/workflows/llm_unit_tests.yml vendored
View file

@ -209,12 +209,12 @@ jobs:
fail-fast: false
matrix:
include:
# - pytorch-version: "2.1"
# python-version: "3.9"
# - pytorch-version: "2.1"
# python-version: "3.10"
# - pytorch-version: "2.1"
# python-version: "3.11"
- pytorch-version: "2.1"
python-version: "3.9"
- pytorch-version: "2.1"
python-version: "3.10"
- pytorch-version: "2.1"
python-version: "3.11"
- pytorch-version: "2.0"
python-version: "3.9"
- pytorch-version: "2.0"
@ -266,9 +266,9 @@ jobs:
- name: Test installed xpu version
shell: bash
run: |
# TODO: differentiate oneapi path after oneapi 2024.0 is installed on test machine
# Specific oneapi position on arc ut test machines
if [[ '${{ matrix.pytorch-version }}' == '2.1' ]]; then
source /opt/intel/oneapi/setvars.sh
source /home/arda/intel/oneapi/setvars.sh
elif [[ '${{ matrix.pytorch-version }}' == '2.0' ]]; then
source /opt/intel/oneapi/setvars.sh
fi
@ -312,9 +312,9 @@ jobs:
- name: Run LLM inference test
shell: bash
run: |
# TODO: differentiate oneapi path after oneapi 2024.0 is installed on test machine
# Specific oneapi position on arc ut test machines
if [[ '${{ matrix.pytorch-version }}' == '2.1' ]]; then
source /opt/intel/oneapi/setvars.sh
source /home/arda/intel/oneapi/setvars.sh
elif [[ '${{ matrix.pytorch-version }}' == '2.0' ]]; then
source /opt/intel/oneapi/setvars.sh
fi
@ -326,9 +326,9 @@ jobs:
run: |
python -m pip install transformers==4.34.0 peft==0.5.0 accelerate==0.23.0
python -m pip install bitsandbytes scipy
# TODO: differentiate oneapi path after oneapi 2024.0 is installed on test machine
# Specific oneapi position on arc ut test machines
if [[ '${{ matrix.pytorch-version }}' == '2.1' ]]; then
source /opt/intel/oneapi/setvars.sh
source /home/arda/intel/oneapi/setvars.sh
elif [[ '${{ matrix.pytorch-version }}' == '2.0' ]]; then
source /opt/intel/oneapi/setvars.sh
fi