Skip to content

GitLab

Unverified Commit 48f589e1 authored by youkaichao's avatar youkaichao Committed by GitHub
Browse files

[mis] fix flaky test of test_cuda_device_count_stateless (#5546)

parent 348616ac
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tests/distributed/test_utils.py
View file @ 48f589e1
...@@ -23,7 +23,8 @@ def test_cuda_device_count_stateless(): ...@@ -23,7 +23,8 @@ def test_cuda_device_count_stateless():
CUDA_VISIBLE_DEVICES is changed.""" CUDA_VISIBLE_DEVICES is changed."""
actor = _CUDADeviceCountStatelessTestActor.options(num_gpus=2).remote() actor = _CUDADeviceCountStatelessTestActor.options(num_gpus=2).remote()
assert ray.get(actor.get_cuda_visible_devices.remote()) == "0,1" assert sorted(ray.get(
actor.get_cuda_visible_devices.remote()).split(",")) == ["0", "1"]
assert ray.get(actor.get_count.remote()) == 2 assert ray.get(actor.get_count.remote()) == 2
ray.get(actor.set_cuda_visible_devices.remote("0")) ray.get(actor.set_cuda_visible_devices.remote("0"))
assert ray.get(actor.get_count.remote()) == 1 assert ray.get(actor.get_count.remote()) == 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