Skip to content

Commit 4c7b755

Browse files
test
1 parent fc3746d commit 4c7b755

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

.github/workflows/tests.yml

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -231,12 +231,12 @@ jobs:
231231
torch_version: ["2.6.0"]
232232
runs-on:
233233
group: bandb-itac-bmemr-gaudi3-1gaudi
234-
container:
235-
image: vault.habana.ai/gaudi-docker/1.20.1/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest
236-
options: --runtime=habana --shm-size=64G --env HABANA_VISIBLE_DEVICES --env HABANA_VISIBLE_MODULES
237-
env:
238-
OMPI_MCA_btl_vader_single_copy_mechanism: none
239-
BNB_TEST_DEVICE: hpu
234+
# container:
235+
# image: vault.habana.ai/gaudi-docker/1.20.1/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest
236+
# options: --runtime=habana --shm-size=64G --env HABANA_VISIBLE_DEVICES --env HABANA_VISIBLE_MODULES
237+
# env:
238+
# OMPI_MCA_btl_vader_single_copy_mechanism: none
239+
# BNB_TEST_DEVICE: hpu
240240
steps:
241241
- name: Show system information
242242
run: |
@@ -246,7 +246,7 @@ jobs:
246246
247247
- name: Show HPU Information
248248
run: |
249-
hpu-smi
249+
hl-smi
250250
251251
test-xpu:
252252
if: github.repository == 'bitsandbytes-foundation/bitsandbytes'

0 commit comments

Comments
 (0)