NUM_RANKS = 2

Author: madsbk

python/cudf_polars/tests/conftest.py

@@ -35,7 +35,7 @@
 # Number of ranks for multi-rank streaming engines that share one GPU
 # (currently ``RayEngine``). Single-GPU dev hosts and CI runners require
 # ``allow_gpu_sharing=True`` to oversubscribe one device across actors.
-NUM_RANKS = 1
+NUM_RANKS = 2
 
 
 @pytest.fixture(params=[False, True], ids=["no_nulls", "nulls"], scope="session")

python/cudf_polars/tests/experimental/test_all_gather_host_data.py

@@ -59,8 +59,6 @@ def test_gather_cluster_info(streaming_engine) -> None:
         assert isinstance(info.gpu_uuid, str)
     # Each rank runs in its own process.
     assert len({info.pid for info in infos}) == streaming_engine.nranks
-    # Without allow_gpu_sharing, all UUIDs must be unique (enforced at init).
-    assert len({info.gpu_uuid for info in infos}) == streaming_engine.nranks
 
 
 def test_cluster_info_cuda_visible_devices(monkeypatch) -> None: