diff --git a/.azure/hpu-tests.yml b/.azure/hpu-tests.yml index 6b153644..53647bce 100644 --- a/.azure/hpu-tests.yml +++ b/.azure/hpu-tests.yml @@ -115,6 +115,7 @@ jobs: tests/test_pytorch/test_datamodule.py \ tests/test_pytorch/test_profiler.py \ tests/test_pytorch/test_precision.py \ + tests/test_pytorch/test_checkpointing.py \ tests/test_pytorch/strategies/test_hpu_parallel.py \ tests/test_pytorch/strategies/test_hpu_ddp.py \ --hpus 1 -W ignore::FutureWarning -m "not standalone_only" \ @@ -155,6 +156,7 @@ jobs: bash tests/run_standalone_tests.sh --hpus 1 -m standalone_only -f \ tests/test_pytorch/strategies/test_hpu_parallel.py \ tests/test_pytorch/test_precision.py \ + tests/test_pytorch/test_checkpointing.py \ tests/test_pytorch/test_dynamic_shapes.py displayName: Standalone-only single card tests