HABANA_VISIBLE_MODULES

huggingface · Feb 18, 2025 · 017f899 · 017f899
1 parent 3ad3e51
commit 017f899
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/ci/accelerate.sh b/tests/ci/accelerate.sh
@@ -1,14 +1,18 @@
 # Description: Accelerate CI script
 
+hl-smi
+echo "HABANA_VISIBLE_DEVICES=${HABANA_VISIBLE_DEVICES}"
+echo "HABANA_VISIBLE_MODULES=${HABANA_VISIBLE_MODULES}"
+
 git clone https://github.com/huggingface/accelerate.git --branch hpu-support --depth 1 && cd accelerate && pip install -e .[testing] git+https://github.com/HabanaAI/[email protected]
 
-RUN_SLOW=1 PT_ENABLE_INT64_SUPPORT=1 make test_cli
+RUN_SLOW=1 PT_ENABLE_INT64_SUPPORT=1 HABANA_VISIBLE_MODULES="" make test_cli
 #       PT_ENABLE_INT64_SUPPORT: 1 # for tokenizers
 if [ $? -ne 0 ]; then
     exit 1
 fi
 
-RUN_SLOW=1 PT_ENABLE_INT64_SUPPORT=1 PT_HPU_LAZY_MODE=0 make test_core
+RUN_SLOW=1 PT_ENABLE_INT64_SUPPORT=1 PT_HPU_LAZY_MODE=0 HABANA_VISIBLE_MODULES="" make test_core
 #      PT_ENABLE_INT64_SUPPORT: 1 # for tokenizers
 #      PT_HPU_LAZY_MODE: 0 # for fsdp
 if [ $? -ne 0 ]; then