From 854db60269cb73b8a41bd6d336471b5361fd94ec Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Carlos=20Mochol=C3=AD?= Date: Wed, 1 Feb 2023 13:23:48 +0100 Subject: [PATCH] Only run TPU standalone tests (#16586) --- .github/workflows/tpu-tests.yml | 2 +- dockers/base-xla/tpu_workflow_fabric.jsonnet | 2 +- dockers/base-xla/tpu_workflow_pytorch.jsonnet | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/tpu-tests.yml b/.github/workflows/tpu-tests.yml index 28f4f7e56b..3f67837dce 100644 --- a/.github/workflows/tpu-tests.yml +++ b/.github/workflows/tpu-tests.yml @@ -41,7 +41,7 @@ jobs: env: PYTHON_VER: 3.8 strategy: - fail-fast: false + fail-fast: true max-parallel: 1 # run sequential matrix: # TODO: add also lightning diff --git a/dockers/base-xla/tpu_workflow_fabric.jsonnet b/dockers/base-xla/tpu_workflow_fabric.jsonnet index 21586ee12a..63845441b7 100644 --- a/dockers/base-xla/tpu_workflow_fabric.jsonnet +++ b/dockers/base-xla/tpu_workflow_fabric.jsonnet @@ -53,7 +53,7 @@ local tputests = base.BaseTest { PL_RUN_TPU_TESTS=1 coverage run --source=lightning_fabric -m pytest -vv --durations=0 ./ echo "--- Running standalone Fabric tests ---" - PL_STANDALONE_TESTS_SOURCE=lightning_fabric PL_STANDALONE_TESTS_BATCH_SIZE=1 bash run_standalone_tests.sh + PL_RUN_TPU_TESTS=1 PL_STANDALONE_TESTS_SOURCE=lightning_fabric PL_STANDALONE_TESTS_BATCH_SIZE=1 bash run_standalone_tests.sh echo "--- Generating coverage ---" coverage xml diff --git a/dockers/base-xla/tpu_workflow_pytorch.jsonnet b/dockers/base-xla/tpu_workflow_pytorch.jsonnet index 5acaf5ce99..5a504e5bf8 100644 --- a/dockers/base-xla/tpu_workflow_pytorch.jsonnet +++ b/dockers/base-xla/tpu_workflow_pytorch.jsonnet @@ -53,7 +53,7 @@ local tputests = base.BaseTest { PL_RUN_TPU_TESTS=1 coverage run --source=pytorch_lightning -m pytest -vv --durations=0 ./ echo "--- Running standalone PL tests ---" - PL_STANDALONE_TESTS_SOURCE=pytorch_lightning PL_STANDALONE_TESTS_BATCH_SIZE=1 bash run_standalone_tests.sh + PL_RUN_TPU_TESTS=1 PL_STANDALONE_TESTS_SOURCE=pytorch_lightning PL_STANDALONE_TESTS_BATCH_SIZE=1 bash run_standalone_tests.sh echo "--- Generating coverage ---" coverage xml