From 8321a6c8e54f2b4212eca3fc31ff573f6da8edd2 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Wed, 9 Oct 2024 11:57:41 +0200 Subject: [PATCH] Try no devices ? --- .github/workflows/build.yaml | 4 ++-- integration-tests/conftest.py | 5 ++++- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 92af2ce6..a48233dd 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -73,11 +73,11 @@ jobs: intel-cpu) export dockerfile="Dockerfile_intel" export label_extension="-intel-cpu" - export docker_devices="" + export docker_devices="none" export docker_volume="/mnt/cache" export runs_on="aws-highmemory-32-plus-priv" export platform="cpu" - export extra_pytest="" + export extra_pytest="-k test_flash_gemma_gptq_load" ;; esac echo $dockerfile diff --git a/integration-tests/conftest.py b/integration-tests/conftest.py index 21b3fd42..e3caadd2 100644 --- a/integration-tests/conftest.py +++ b/integration-tests/conftest.py @@ -507,7 +507,10 @@ def launcher(event_loop): volumes = [f"{DOCKER_VOLUME}:/data"] if DOCKER_DEVICES is not None: - devices = DOCKER_DEVICES.strip().split(",") + if DOCKER_DEVICES.lower() == "none": + devices = [] + else: + devices = DOCKER_DEVICES.strip().split(",") visible = os.getenv("ROCR_VISIBLE_DEVICES") if visible: env["ROCR_VISIBLE_DEVICES"] = visible