Fix(CI): fix little bug in yaml once more (#191)

InternLM · Apr 10, 2024 · 985681f · 985681f
1 parent 27eb3b3
commit 985681f
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/.github/workflows/pr_before_merge.yaml b/.github/workflows/pr_before_merge.yaml
@@ -59,8 +59,6 @@ jobs:
         echo "::add-mask::${{env.WORKSPACE_PREFIX}}"
         echo "::add-mask::$path_prefix"
     - uses: actions/checkout@v3
-      with:
-          ref: ${{ github.event_name == 'schedule' && 'develop' || github.event_name == 'workflow_dispatch' && '' }}
 
     - name: test_embedding_accuracy
       run: |

diff --git a/.github/workflows/pr_merged.yaml b/.github/workflows/pr_merged.yaml
@@ -27,7 +27,7 @@ jobs:
         source activate ${evo_env_torch21_flash2}
         export PYTHONPATH=$PWD:$PYTHONPATH
         jobname=${GITHUB_RUN_ID}-${GITHUB_JOB}-${GITHUB_RUN_ATTEMPT}
-        srun -p ${SLURM_PARTITION} --kill-on-bad-exit=1 --job-name=$jobname -N 1 -n 8 --ntasks-per-node=8 --gpus-per-task=1  python ./tests/test_training/train_CI.py --config ./tests/test_training/7B_check_acc.py
+        srun -p ${SLURM_PARTITION} --kill-on-bad-exit=1 --job-name=$jobname -N 1 -n 8 --ntasks-per-node=8 --gpus-per-task=1 --exclusive python ./tests/test_training/train_CI.py --config ./tests/test_training/7B_check_acc.py
         exit_code=$?
         sh ./ci_scripts/common/check_slurm_cancled.sh $exit_code $jobname
 

diff --git a/tests/test_model/test_norm.py b/tests/test_model/test_norm.py
@@ -14,7 +14,7 @@ def check_norm(args):
     # init
     rank, world_size = args
     build_environment(rank, world_size)
-    device = get_current_device(rank)
+    device = get_current_device()
     rtol, atol = (1e-3, 5e-3)
     hidden_size = 4
     layer_norm_epsilon = 1e-05