Release gil when doing ray.wait. (#4190)

2026-06-28 07:34:30 +08:00 · 2019-02-28 00:32:07 -08:00
parent 484708d44d
commit 9c5fdbb63c
1 changed files with 5 additions and 3 deletions
@@ -263,9 +263,11 @@ cdef class RayletClient:
            WaitResultPair result
            c_vector[CObjectID] wait_ids
        wait_ids = ObjectIDsToVector(object_ids)
-        check_status(self.client.get().Wait(wait_ids, num_returns,
-                                            timeout_milliseconds, wait_local,
-                                            current_task_id.data, &result))
+        with nogil:
+            check_status(self.client.get().Wait(wait_ids, num_returns,
+                                                timeout_milliseconds,
+                                                wait_local,
+                                                current_task_id.data, &result))
        return (VectorToObjectIDs(result.first),
                VectorToObjectIDs(result.second))