From 0a87c4dbdd780e787ca6640b139970297e837359 Mon Sep 17 00:00:00 2001 From: Terry Kong Date: Tue, 4 Jun 2024 13:59:41 -0700 Subject: [PATCH] Reduce number of t5x tests --- .github/workflows/_test_t5x_rosetta.yaml | 14 -------- .github/workflows/_test_upstream_t5x.yaml | 44 ----------------------- 2 files changed, 58 deletions(-) diff --git a/.github/workflows/_test_t5x_rosetta.yaml b/.github/workflows/_test_t5x_rosetta.yaml index 7d5c03b29..3c94e2ea6 100644 --- a/.github/workflows/_test_t5x_rosetta.yaml +++ b/.github/workflows/_test_t5x_rosetta.yaml @@ -38,10 +38,6 @@ jobs: strategy: matrix: include: - - TEST_NAME: "1P1G_te-1" - N_GPU: 1 - ADDITIONAL_ARGS: "" - EXTRA_GIN_ARGS: "--gin.train/utils.DatasetConfig.pack=False --gin.train_eval/utils.DatasetConfig.pack=False" - TEST_NAME: "1P1G_te-0" N_GPU: 1 ADDITIONAL_ARGS: "--enable-te 0" @@ -234,16 +230,6 @@ jobs: strategy: matrix: include: - - TEST_NAME: "1N1G-te-1" - N_GPU: 1 - N_NODE: 1 - ADDITIONAL_ARGS: "" - EXTRA_GIN_ARGS: "--gin.train/utils.DatasetConfig.pack=False --gin.train_eval/utils.DatasetConfig.pack=False" - - TEST_NAME: "1N8G-te-1" - N_GPU: 8 - N_NODE: 1 - ADDITIONAL_ARGS: "" - EXTRA_GIN_ARGS: "--gin.train/utils.DatasetConfig.pack=False --gin.train_eval/utils.DatasetConfig.pack=False" - TEST_NAME: "2N8G-te-1" N_GPU: 8 N_NODE: 2 diff --git a/.github/workflows/_test_upstream_t5x.yaml b/.github/workflows/_test_upstream_t5x.yaml index c71f90de4..e240a2df3 100644 --- a/.github/workflows/_test_upstream_t5x.yaml +++ b/.github/workflows/_test_upstream_t5x.yaml @@ -39,20 +39,8 @@ jobs: strategy: matrix: include: - - TEST_NAME: "1P1G" - N_GPU: 1 - ADDITIONAL_ARGS: "" - - TEST_NAME: "1P2G" - N_GPU: 2 - ADDITIONAL_ARGS: "" - - TEST_NAME: "1P4G" - N_GPU: 4 - ADDITIONAL_ARGS: "" - TEST_NAME: "1P8G" N_GPU: 8 - - TEST_NAME: "1P1G_fmha" - N_GPU: 1 - ADDITIONAL_ARGS: "--enable-fmha 1" - TEST_NAME: "1P2G_fmha" N_GPU: 2 ADDITIONAL_ARGS: "--enable-fmha 1" @@ -193,42 +181,10 @@ jobs: strategy: matrix: include: - - TEST_NAME: "1G1N" - N_GPU: 1 - N_NODE: 1 - ADDITIONAL_ARGS: "" - - TEST_NAME: "2G1N" - N_GPU: 2 - N_NODE: 1 - ADDITIONAL_ARGS: "" - - TEST_NAME: "4G1N" - N_GPU: 4 - N_NODE: 1 - ADDITIONAL_ARGS: "" - - TEST_NAME: "8G1N" - N_GPU: 8 - N_NODE: 1 - ADDITIONAL_ARGS: "" - - TEST_NAME: "1G2N" - N_GPU: 1 - N_NODE: 2 - ADDITIONAL_ARGS: "" - - TEST_NAME: "2G2N" - N_GPU: 2 - N_NODE: 2 - ADDITIONAL_ARGS: "" - - TEST_NAME: "4G2N" - N_GPU: 4 - N_NODE: 2 - ADDITIONAL_ARGS: "" - TEST_NAME: "8G2N" N_GPU: 8 N_NODE: 2 ADDITIONAL_ARGS: "" - - TEST_NAME: "2G2N_fmha" - N_GPU: 2 - N_NODE: 2 - ADDITIONAL_ARGS: "--enable-fmha 1" - TEST_NAME: "8G2N_fmha" N_GPU: 8 N_NODE: 2