From acf5402e59e31b1dd1cdaff7a37a27f4aab4ce17 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Mon, 16 Sep 2024 10:54:49 +0530 Subject: [PATCH 1/2] handle DeprecatedAttentionBlockTests. --- .github/workflows/push_tests.yml | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/.github/workflows/push_tests.yml b/.github/workflows/push_tests.yml index a92507d1b215..d55b6e11bb84 100644 --- a/.github/workflows/push_tests.yml +++ b/.github/workflows/push_tests.yml @@ -140,7 +140,15 @@ jobs: # https://pytorch.org/docs/stable/notes/randomness.html#avoiding-nondeterministic-algorithms CUBLAS_WORKSPACE_CONFIG: :16:8 run: | - python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile \ + # DeprecatedAttentionBlockTests::test_conversion_when_using_device_map fails + # without this. `DeprecatedAttentionBlockTests` aren't used anyway, so this should + # be okay. + if [[ "${{ matrix.module }}" != "models" ]]; then + PYTEST_ARGS="-n 1 --dist=loadfile" + else + PYTEST_ARGS="" + fi + python -m pytest $PYTEST_ARGS --max-worker-restart=0 \ -s -v -k "not Flax and not Onnx" \ --make-reports=tests_torch_cuda \ tests/${{ matrix.module }} From e9658e844730ea509d6a4148fba24c2f13a9965e Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Mon, 16 Sep 2024 13:29:51 +0530 Subject: [PATCH 2/2] fix for release tests --- .github/workflows/release_tests_fast.yml | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/.github/workflows/release_tests_fast.yml b/.github/workflows/release_tests_fast.yml index 081e90585ef5..cf6dd5ad87fd 100644 --- a/.github/workflows/release_tests_fast.yml +++ b/.github/workflows/release_tests_fast.yml @@ -139,6 +139,14 @@ jobs: # https://pytorch.org/docs/stable/notes/randomness.html#avoiding-nondeterministic-algorithms CUBLAS_WORKSPACE_CONFIG: :16:8 run: | + # DeprecatedAttentionBlockTests::test_conversion_when_using_device_map fails + # without this. `DeprecatedAttentionBlockTests` aren't used anyway, so this should + # be okay. + if [[ "${{ matrix.module }}" != "models" ]]; then + PYTEST_ARGS="-n 1 --dist=loadfile" + else + PYTEST_ARGS="" + fi python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile \ -s -v -k "not Flax and not Onnx" \ --make-reports=tests_torch_cuda \