From a3d0799963e5c9468bbf6336d1e2b9a63d998abd Mon Sep 17 00:00:00 2001 From: Saeed Maleki <30272783+saeedmaleki@users.noreply.github.com> Date: Sun, 18 Feb 2024 18:39:21 -0800 Subject: [PATCH] Fix the comm.py for nvls (#267) Fix the comm.py for nvls --- python/mscclpp/comm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/mscclpp/comm.py b/python/mscclpp/comm.py index b4f11ae4..b3cc51f1 100644 --- a/python/mscclpp/comm.py +++ b/python/mscclpp/comm.py @@ -97,7 +97,7 @@ class CommGroup: else: endpoint = endpoints if endpoint.transport == Transport.Nvls: - connections[rank] = self.communicator.connct_nvls_collective(all_ranks, endpoint) + return self.communicator.connct_nvls_collective(all_ranks, endpoint) else: connections[rank] = self.communicator.connect_on_setup(rank, 0, endpoint) self.communicator.setup()