diff --git a/.azure/hpu-tests.yml b/.azure/hpu-tests.yml index 0fa04920ec7cc..f0b279bda3f60 100644 --- a/.azure/hpu-tests.yml +++ b/.azure/hpu-tests.yml @@ -53,6 +53,11 @@ jobs: FREEZE_REQUIREMENTS: 1 displayName: 'Install dependencies' + - bash: | + hl-smi -L + lsmod | grep habanalabs + displayName: 'Check the driver status' + - bash: | python -m pytest -sv accelerators/test_hpu.py --forked --junitxml=hpu1_test-results.xml workingDirectory: tests/tests_pytorch