Update fork-maintenance.yml

Cemberk · Aug 23, 2024 · 346cdd0 · 346cdd0
1 parent 49cc7bd
commit 346cdd0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/fork-maintenance.yml b/.github/workflows/fork-maintenance.yml
@@ -34,5 +34,5 @@ jobs:
               ]
             }
           pr_branch_prefix: "scheduled-merge"
-          unit_test_command: ""
-          performance_test_command: ""
+          unit_test_command: "machine_type=$(/opt/rocm/bin/rocminfo | grep -o -m 1 'gfx.*[^ ]'); cd tests; folders=($(python3 -c 'import os; tests = os.getcwd(); model_tests = os.listdir(os.path.join(tests, "models")); d1 = sorted(list(filter(os.path.isdir, os.listdir(tests)))); d2 = sorted(list(filter(os.path.isdir, [f"models/{x}" for x in model_tests]))); d1.remove("models"); d = d2 + d1; print("\n".join(d))')); cd ..; for folder in "${folders[@]}"; do pytest tests/${folder} -v --make-reports="huggingface_unit_tests_${machine_type}_run_models_gpu_${folder}" -rfEsx --continue-on-collection-errors -m "not not_device_test" -p no:cacheprovider; done; allstats=($(find reports -name stats.txt)); for stat in "${allstats[@]}"; do echo "$stat"; cat $stat; done"
+          performance_test_command: "python examples/pytorch/language-modeling/run_mlm.py --model_name_or_path bert-base-uncased --dataset_name wikitext --dataset_config_name wikitext-2-raw-v1 --do_train --do_eval --output_dir /tmp/test-mlm --per_device_train_batch_size 8 --per_device_eval_batch_size 8 --max_steps 500"