From 3ed877adbf7db873a8adb83d56401ffd3753a2aa Mon Sep 17 00:00:00 2001 From: lucasdelimanogueira Date: Fri, 24 May 2024 20:16:08 -0300 Subject: [PATCH] fix distributed allreduce calling python --- norch/distributed/distributed.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/norch/distributed/distributed.py b/norch/distributed/distributed.py index b078664..9c797f8 100644 --- a/norch/distributed/distributed.py +++ b/norch/distributed/distributed.py @@ -14,11 +14,11 @@ def broadcast_tensor(tensor): Tensor._C.broadcast_tensor.argtypes = [ctypes.POINTER(CTensor)] Tensor._C.broadcast_tensor.restype = None - Tensor._C.broadcast_tensor(tensor) + Tensor._C.broadcast_tensor(tensor.tensor) def allreduce_sum_tensor(tensor): Tensor._C.allreduce_sum_tensor.argtypes = [ctypes.POINTER(CTensor)] Tensor._C.allreduce_sum_tensor.restype = None - Tensor._C.allreduce_sum_tensor(tensor) + Tensor._C.allreduce_sum_tensor(tensor.tensor)