Skip to content

Commit 1364a14

Browse files
authored
[core][rdt] Fix flaky send fail gloo test (#58924)
Signed-off-by: dayshah <dhyey2019@gmail.com>
1 parent 6403675 commit 1364a14

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

python/ray/tests/gpu_objects/test_gpu_objects_gloo.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@ def send(self, tensor):
6969
def recv(self, tensor):
7070
return tensor
7171

72+
@ray.method(concurrency_group="_ray_system")
7273
def clear_gpu_object_store(self):
7374
gpu_object_store = (
7475
ray._private.worker.global_worker.gpu_object_manager.gpu_object_store

0 commit comments

Comments
 (0)