Skip to content

Commit

Permalink
Merge branch 'olruwase/ds_5241' of github.com:microsoft/DeepSpeed int…
Browse files Browse the repository at this point in the history
…o olruwase/ds_5241
  • Loading branch information
tjruwase committed Feb 6, 2025
2 parents 2c6f630 + cccd5b1 commit 21bfca0
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 7 deletions.
19 changes: 13 additions & 6 deletions .github/workflows/xpu-max1100.yml
Original file line number Diff line number Diff line change
Expand Up @@ -36,26 +36,31 @@ jobs:
unit-tests:
runs-on: [self-hosted, intel, xpu]
container:
image: intel/oneapi-basekit:2024.2.1-0-devel-ubuntu22.04
image: intel/oneapi-basekit:2025.0.1-0-devel-ubuntu24.04
ports:
- 80
options: --privileged -it --rm --device /dev/dri:/dev/dri -v /dev/dri/by-path:/dev/dri/by-path --ipc=host --cap-add=ALL

steps:
- uses: actions/checkout@v4
- name: Install prerequisite
shell: bash
run: |
apt-get update
apt-get install clinfo libaio-dev python3-pip -y
pip install torch==2.3.1 -f https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/torch/
pip install intel-extension-for-pytorch==2.3.110+xpu -f https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/intel-extension-for-pytorch/
pip install oneccl_bind_pt==2.3.100+xpu -f https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/oneccl-bind-pt/
pip install torchvision==0.18.1 -f https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/torchvision/
apt-get install clinfo libaio-dev python3-pip python3.12-venv -y
python3 -m venv ~/ds_env
source ~/ds_env/bin/activate
pip install torch==2.5.1 -f https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/torch/
pip install intel-extension-for-pytorch==2.5.10+xpu -f https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/intel-extension-for-pytorch/
pip install oneccl_bind_pt==2.5.0+xpu -f https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/oneccl-bind-pt/
pip install torchvision==0.20.1 -f https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/torchvision/
pip install py-cpuinfo numpy
pip install .[dev,autotuning]
- name: Check container state
shell: bash
run: |
source ~/ds_env/bin/activate
ldd --version
ds_report
python3 -c "import torch; print('torch:', torch.__version__, torch)"
Expand All @@ -64,7 +69,9 @@ jobs:
pip list
- name: Unit tests
shell: bash
run: |
source ~/ds_env/bin/activate
cd tests/unit
pytest --verbose accelerator/*
pytest --verbose autotuning/*
Expand Down
2 changes: 1 addition & 1 deletion docs/CNAME
Original file line number Diff line number Diff line change
@@ -1 +1 @@
www.deepspeed.ai
www.deepspeed.ai

0 comments on commit 21bfca0

Please sign in to comment.