From 55cdfbfce3d533f73366556171cb4ea6de858aea Mon Sep 17 00:00:00 2001 From: baptiste Date: Wed, 21 May 2025 15:28:58 +0000 Subject: [PATCH] enable multi-card test --- .github/workflows/build.yaml | 2 +- integration-tests/gaudi/test_gaudi_generate.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index c7ccf764..14c69a2b 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -129,7 +129,7 @@ jobs: export label_extension="-gaudi" export docker_volume="/mnt/cache" export docker_devices="" - export runs_on="itac-bm-emr-gaudi3-dell-1gaudi" + export runs_on="itac-bm-emr-gaudi3-dell-8gaudi" export platform="" export extra_pytest="--gaudi" export target="" diff --git a/integration-tests/gaudi/test_gaudi_generate.py b/integration-tests/gaudi/test_gaudi_generate.py index 26ba47fe..c2d768d0 100644 --- a/integration-tests/gaudi/test_gaudi_generate.py +++ b/integration-tests/gaudi/test_gaudi_generate.py @@ -32,6 +32,7 @@ TEST_CONFIGS = { "--max-batch-prefill-tokens", "2048", ], + "run_by_default": True, }, "meta-llama/Llama-3.1-8B-Instruct": { "model_id": "meta-llama/Llama-3.1-8B-Instruct",