File tree Expand file tree Collapse file tree 1 file changed +40
-2
lines changed Expand file tree Collapse file tree 1 file changed +40
-2
lines changed Original file line number Diff line number Diff line change @@ -161,6 +161,46 @@ jobs:
161161 - name : Run tests
162162 run : pytest --durations=100
163163
164+ test-cpu-ipex :
165+ if : github.repository == 'bitsandbytes-foundation/bitsandbytes'
166+ needs : build-cpu
167+ runs-on : banb-aws-general-8-plus-use1-public-80
168+ env :
169+ BNB_TEST_DEVICE : cpu
170+ steps :
171+ - uses : actions/checkout@v4
172+
173+ - name : Download build artifact
174+ uses : actions/download-artifact@v4
175+ with :
176+ name : lib_cpu_ubuntu-22.04_x86_64
177+ path : bitsandbytes/
178+ merge-multiple : true
179+
180+ - name : Setup Python
181+ uses : actions/setup-python@v5
182+ with :
183+ python-version : 3.9
184+
185+ - name : Install dependencies
186+ run : |
187+ pip install torch==2.7.0 --index-url https://download.pytorch.org/whl/cpu
188+ pip install intel_extension_for_pytorch==2.7.0 -extra-index-url https://pytorch-extension.intel.com/release-whl/stable/cpu/us/
189+ pip install -e ".[test]"
190+ pip install pytest-cov
191+
192+ - name : Show installed packages
193+ run : pip list
194+
195+ - name : Show environment information
196+ run : python -m torch.utils.collect_env
197+
198+ - name : IPEX smoke test
199+ run : python -c "import torch; import intel_extension_for_pytorch as ipex; print(torch.__version__); print(ipex.__version__);"
200+
201+ - name : Run tests
202+ run : pytest --durations=100
203+
164204 # test-cuda-aarch64:
165205 # if: github.repository == 'bitsandbytes-foundation/bitsandbytes'
166206 # needs: build-cuda
@@ -182,8 +222,6 @@ jobs:
182222 # - name: Show pip packages
183223 # run: pip list
184224
185-
186-
187225 test-cuda :
188226 if : github.repository == 'bitsandbytes-foundation/bitsandbytes'
189227 needs : build-cuda
You can’t perform that action at this time.
0 commit comments