Skip to content
This repository was archived by the owner on Jul 24, 2024. It is now read-only.

Commit 98ea94c

Browse files
committed
Changed ipex from ipex-xpu -> ipex-cpu (#90)
1 parent 41d4661 commit 98ea94c

File tree

2 files changed

+25
-3
lines changed

2 files changed

+25
-3
lines changed

tests/conda-envs/ipex-xpu.yaml

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
name: ipex
2+
channels:
3+
- intel
4+
- conda-forge
5+
dependencies:
6+
- intel-aikit-pytorch
7+
- pytorch>=2.0.1=*_xpu_*
8+
- intel-extension-for-pytorch
9+
- datasets
10+
- accelerate
11+
- sentencepiece
12+
# The following packages are required to run benchmarks
13+
- sqlalchemy>=2.0.0
14+
- pytest

tests/conda-envs/ipex.yaml

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,20 @@ channels:
33
- intel
44
- conda-forge
55
dependencies:
6-
- intel-aikit-pytorch
7-
- pytorch>=2.0.1=*_xpu_*
8-
- intel-extension-for-pytorch
6+
- python=3.11
97
- datasets
108
- accelerate
119
- sentencepiece
1210
# The following packages are required to run benchmarks
1311
- sqlalchemy>=2.0.0
1412
- pytest
13+
- pip
14+
- pip:
15+
- --extra-index-url https://download.pytorch.org/whl/cpu
16+
- torch
17+
- torchvision
18+
- torchaudio
19+
- transformers==4.35.2
20+
- intel-extension-for-pytorch
21+
- --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/cpu/us/
22+
- oneccl_bind_pt

0 commit comments

Comments
 (0)