Skip to content

GitLab

Unverified Commit 292fc59d authored by Michael Goin's avatar Michael Goin Committed by GitHub
Browse files

[CI] Actually run tests/kv_transfer/test_disagg.py in CI (#17555) 


Signed-off-by: default avatarmgoin <mgoin64@gmail.com>
parent afcb3f88
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/kv_transfer/test_disagg.py
View file @ 292fc59d
...@@ -14,8 +14,8 @@ import torch ...@@ -14,8 +14,8 @@ import torch
# Fixture to set up environment variables and teardown servers after tests # Fixture to set up environment variables and teardown servers after tests
@pytest.fixture(scope="module", autouse=True) @pytest.fixture(scope="module", autouse=True)
def setup_servers(): def setup_servers():
if torch.cuda.device_count() < 4: if torch.cuda.device_count() < 2:
pytest.skip("Skipping test: fewer than 4 GPUs available") pytest.skip("Skipping test: fewer than 2 GPUs available")
# Set up environment variables # Set up environment variables
VLLM_HOST_IP = subprocess.check_output("hostname -I | awk '{print $1}'", VLLM_HOST_IP = subprocess.check_output("hostname -I | awk '{print $1}'",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment