File tree Expand file tree Collapse file tree 1 file changed +7
-7
lines changed Expand file tree Collapse file tree 1 file changed +7
-7
lines changed Original file line number Diff line number Diff 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'
You can’t perform that action at this time.
0 commit comments