From 987c675d6666a06e95e2684aa432df1deb2dd545 Mon Sep 17 00:00:00 2001 From: Cemberk Date: Tue, 10 Sep 2024 15:54:15 -0500 Subject: [PATCH] Update fork-maintenance.yml --- .github/workflows/fork-maintenance.yml | 15 +-------------- 1 file changed, 1 insertion(+), 14 deletions(-) diff --git a/.github/workflows/fork-maintenance.yml b/.github/workflows/fork-maintenance.yml index 6cfe2d4fc646..a90a46ec39bc 100644 --- a/.github/workflows/fork-maintenance.yml +++ b/.github/workflows/fork-maintenance.yml @@ -43,20 +43,7 @@ jobs: pip install huggingface_hub datasets ; pip install parameterized; unit_test_command: > - 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, [fmodels{x} for x in model_tests]))); - d1.remove(models); - d = d2 + d1; - print(n.join(d))' )); - cd ..; - echo "before run"; - 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 nocacheprovider ; done ; - echo "after run "; - allstats=(\$(find reports -name stats.txt)); for stat in \${allstats[@]}; do echo \$stat; cat \$stat; done; - echo "after report" + pytest performance_test_command: > echo "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" docker_image: rocm/pytorch:latest