test

tests/ci/accelerate.sh

@@ -23,13 +23,16 @@ pip install -e .[testing]
 pip install git+https://github.com/HabanaAI/[email protected]
 
 # Set environment variables
-export PT_HPU_EAGER_COLLECTIVE_PIPELINE_ENABLE=0
-export PT_HPU_EAGER_PIPELINE_ENABLE=0
 export PT_ENABLE_INT64_SUPPORT=1
 export PT_HPU_LAZY_MODE=0
 export RUN_SLOW=1
 
-pytest tests/ -s -vvvv -k "test_multi_device_merge_fsdp_weights"
+pytest tests/ -s -vvvv -k "test_ddp_comm_hook"
+
+if [ $? -ne 0 ]; then
+    exit 1
+fi
+
 # CORE FEATURES TESTS
 
 echo "Running CLI tests"