From cbb3a97ea22513faed1e7052b8708655b267ee59 Mon Sep 17 00:00:00 2001 From: Param Bole Date: Wed, 15 Jan 2025 15:31:08 -0800 Subject: [PATCH] Fixit: Removing redundant tests (#559) --- dags/sparsity_diffusion_devx/jax_stable_stack_gpu_e2e.py | 9 --------- dags/sparsity_diffusion_devx/project_bite_gpu_e2e.py | 1 - 2 files changed, 10 deletions(-) diff --git a/dags/sparsity_diffusion_devx/jax_stable_stack_gpu_e2e.py b/dags/sparsity_diffusion_devx/jax_stable_stack_gpu_e2e.py index 13c69756..47019eab 100644 --- a/dags/sparsity_diffusion_devx/jax_stable_stack_gpu_e2e.py +++ b/dags/sparsity_diffusion_devx/jax_stable_stack_gpu_e2e.py @@ -68,15 +68,6 @@ for model, (test_script, nnodes) in test_models_gpu.items(): for mode, image in docker_images: - stable_a3_gpu = config.get_gpu_gke_test_config( - time_out_in_min=300, - test_name=f"maxtext-stable-stack-{mode.value}-{model}", - run_model_cmds=(test_script,), - num_slices=nnodes, - cluster=XpkClusters.GPU_A3_CLUSTER, - docker_image=image.value, - test_owner=test_owner.PARAM_B, - ).run_with_quarantine(quarantine_task_group) stable_a3plus_gpu = config.get_gpu_gke_test_config( time_out_in_min=300, test_name=f"maxtext-stable-stack-{mode.value}-{model}", diff --git a/dags/sparsity_diffusion_devx/project_bite_gpu_e2e.py b/dags/sparsity_diffusion_devx/project_bite_gpu_e2e.py index 5c399e7f..0e3600b0 100644 --- a/dags/sparsity_diffusion_devx/project_bite_gpu_e2e.py +++ b/dags/sparsity_diffusion_devx/project_bite_gpu_e2e.py @@ -44,7 +44,6 @@ axlearn_test_configs = { # accelerator: list of slices to test - "a3": [1], "a3plus": [1, 2], }