From d2361d7fd856a8ca0aecd49a9e39a6846cbc183b Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Wed, 9 Oct 2024 12:42:34 +0200 Subject: [PATCH] Temp deactivate intel, activate nvidia ? --- .github/workflows/build.yaml | 3 ++- integration-tests/conftest.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 2e3ce8f1..d0aaea27 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -75,7 +75,8 @@ jobs: export label_extension="-intel-cpu" export docker_devices="none" export docker_volume="/mnt/cache" - export runs_on="aws-highmemory-32-plus-priv" + export runs_on="ubuntu-latest" + # export runs_on="aws-highmemory-32-plus-priv" export platform="cpu" export extra_pytest="-k test_flash_llama_load" ;; diff --git a/integration-tests/conftest.py b/integration-tests/conftest.py index a2cab51f..20521648 100644 --- a/integration-tests/conftest.py +++ b/integration-tests/conftest.py @@ -528,7 +528,7 @@ def launcher(event_loop): for i in range(gpu_count) ] else: - devices = [] + devices = None device_requests = [ docker.types.DeviceRequest(count=gpu_count, capabilities=[["gpu"]]) ]