From 6549ebebdfdf036efc32dc2802d9f31c03dde948 Mon Sep 17 00:00:00 2001 From: wangmengke Date: Fri, 24 Nov 2023 15:33:38 +0800 Subject: [PATCH] change trigger event --- .github/workflows/weekly_test.yaml | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/.github/workflows/weekly_test.yaml b/.github/workflows/weekly_test.yaml index 3c278e6..37a958f 100644 --- a/.github/workflows/weekly_test.yaml +++ b/.github/workflows/weekly_test.yaml @@ -15,6 +15,7 @@ jobs: - name: training_8GPU run: | + git checkout develop && git rev-parse HEAD source /mnt/petrelfs/share_data/llm_env/env/llm-flash2.0 srun -p ${SLURM_PARTITION} --kill-on-bad-exit=1 --job-name=${GITHUB_RUN_ID}-${GITHUB_JOB} -n8 --ntasks-per-node=8 --cpus-per-task=4 --gpus-per-task=1 pytest -s -v --color=yes -m "training_8GPU" ./tests/test_training @@ -26,6 +27,7 @@ jobs: - name: training_16GPU_8DP2TP run: | + git checkout develop && git rev-parse HEAD source /mnt/petrelfs/share_data/llm_env/env/llm-flash2.0 sed -i 's/^.*tensor=.*/ tensor=2,/' ./configs/7B_sft.py srun -p ${SLURM_PARTITION} --kill-on-bad-exit=1 --job-name=${GITHUB_RUN_ID}-${GITHUB_JOB} -n16 --ntasks-per-node=8 --cpus-per-task=4 --gpus-per-task=1 pytest -s -v --color=yes -m "training_16GPU_8DP2TP" ./tests/test_training @@ -38,6 +40,7 @@ jobs: - name: training_16GPU_8DP2TPSP run: | + git checkout develop && git rev-parse HEAD source /mnt/petrelfs/share_data/llm_env/env/llm-flash2.0 sed -i 's/^.*tensor=.*/ tensor=2,/' ./configs/7B_sft.py sed -i 's/^.*sequence_parallel=.*/ sequence_parallel=True,/' ./configs/7B_sft.py @@ -51,6 +54,7 @@ jobs: - name: training_16GPU_8DP2PP run: | + git checkout develop && git rev-parse HEAD source /mnt/petrelfs/share_data/llm_env/env/llm-flash2.0 sed -i 's/^.*pipeline=.*/ pipeline=dict(size=2),/' ./configs/7B_sft.py srun -p ${SLURM_PARTITION} --kill-on-bad-exit=1 --job-name=${GITHUB_RUN_ID}-${GITHUB_JOB} -n16 --ntasks-per-node=8 --cpus-per-task=4 --gpus-per-task=1 pytest -s -v --color=yes -m "training_16GPU_8DP2PP" ./tests/test_training @@ -63,6 +67,7 @@ jobs: - name: training_16GPU_8DP2PP_InterleavedOverlap run: | + git checkout develop && git rev-parse HEAD source /mnt/petrelfs/share_data/llm_env/env/llm-flash2.0 sed -i 's/^.*pipeline=.*/ pipeline=dict(size=2, interleaved_overlap=True),/' ./configs/7B_sft.py sed -i 's/^.*num_chunks=.*/ num_chunks=2,/' ./configs/7B_sft.py @@ -76,6 +81,7 @@ jobs: - name: test_optimizer run: | + git checkout develop && git rev-parse HEAD source /mnt/petrelfs/share_data/llm_env/env/llm-flash2.0 srun -p ${SLURM_PARTITION} --kill-on-bad-exit=1 --job-name=${GITHUB_RUN_ID}-${GITHUB_JOB} -N 1 -n 1 --gres=gpu:8 python -m pytest -s ./tests/test_solver/test_optimizer.py @@ -87,6 +93,7 @@ jobs: - name: test_embedding_accuracy run: | + git checkout develop && git rev-parse HEAD source /mnt/petrelfs/share_data/llm_env/env/llm-flash2.0 srun -p ${SLURM_PARTITION} --kill-on-bad-exit=1 --job-name=${GITHUB_RUN_ID}-${GITHUB_JOB} -N 1 -n 1 --gres=gpu:8 python -m pytest -s ./tests/test_model/test_embedding.py