From acc2b4f8da414d31b46434f5cd465910a636812f Mon Sep 17 00:00:00 2001 From: hariharandev1 Date: Sat, 5 Oct 2024 02:02:51 -0700 Subject: [PATCH] switch order of faster tests --- .github/workflows/ci.yml | 28 ++++++++++++++-------------- 1 file changed, 14 insertions(+), 14 deletions(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 4c90663d..ff052dc9 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -78,6 +78,20 @@ jobs: source ${VENV_PATH}/bin/activate pip install --upgrade pip pip install -r requirements.txt + - name: test_checkpoint_epoch + run: | + source ${VENV_PATH}/bin/activate + mpirun -np 2 pytest -k test_checkpoint_epoch[tensorflow-1024-optimizers0-2-layer_params0-all_ranks] -v + mpirun -np 2 pytest -k test_checkpoint_epoch[pytorch-1024-optimizers1-2-layer_params1-all_ranks] -v + mpirun -np 2 pytest -k test_checkpoint_epoch[tensorflow-1024-optimizers2-2-layer_params2-rank_zero] -v + mpirun -np 2 pytest -k test_checkpoint_epoch[pytorch-1024-optimizers3-2-layer_params3-rank_zero] -v + mpirun -np 2 pytest -k test_checkpoint_epoch[tensorflow-1024-optimizers4-1-layer_params4-all_ranks] -v + mpirun -np 2 pytest -k test_checkpoint_epoch[pytorch-1024-optimizers5-1-layer_params5-all_ranks] -v + rm -rf data + - name: test_checkpoint_step + run: | + source ${VENV_PATH}/bin/activate + mpirun -np 2 pytest -k test_checkpoint_step -v - name: test_gen_data run: | source ${VENV_PATH}/bin/activate @@ -182,20 +196,6 @@ jobs: mpirun -np 2 pytest -k test_custom_storage_root_train[mmap_indexed_binary-tensorflow] -v mpirun -np 2 pytest -k test_custom_storage_root_train[mmap_indexed_binary-pytorch] -v rm -rf data - - name: test_checkpoint_epoch - run: | - source ${VENV_PATH}/bin/activate - mpirun -np 2 pytest -k test_checkpoint_epoch[tensorflow-1024-optimizers0-2-layer_params0-all_ranks] -v - mpirun -np 2 pytest -k test_checkpoint_epoch[pytorch-1024-optimizers1-2-layer_params1-all_ranks] -v - mpirun -np 2 pytest -k test_checkpoint_epoch[tensorflow-1024-optimizers2-2-layer_params2-rank_zero] -v - mpirun -np 2 pytest -k test_checkpoint_epoch[pytorch-1024-optimizers3-2-layer_params3-rank_zero] -v - mpirun -np 2 pytest -k test_checkpoint_epoch[tensorflow-1024-optimizers4-1-layer_params4-all_ranks] -v - mpirun -np 2 pytest -k test_checkpoint_epoch[pytorch-1024-optimizers5-1-layer_params5-all_ranks] -v - rm -rf data - - name: test_checkpoint_step - run: | - source ${VENV_PATH}/bin/activate - mpirun -np 2 pytest -k test_checkpoint_step -v - name: test_eval run: | source ${VENV_PATH}/bin/activate