up

Signed-off-by: Rui Qiao <ruisearch42@gmail.com>
ray-project · Aug 30, 2024 · b1c5bc2 · b1c5bc2
1 parent aa50a56
commit b1c5bc2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/ray/experimental/channel/conftest.py b/python/ray/experimental/channel/conftest.py
@@ -90,6 +90,7 @@ def recv(
         buf = allocator(shape, dtype)
         buf[:] = received_tensor[:]
         self.num_ops[barrier_key] += 1
+        return buf
 
 
 def start_nccl_mock():