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