Fix the comm.py for nvls (#267)

Fix the comm.py for nvls
microsoft · Feb 19, 2024 · a3d0799 · a3d0799
1 parent 5971508
commit a3d0799
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/mscclpp/comm.py b/python/mscclpp/comm.py
@@ -97,7 +97,7 @@ def make_connection(
             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()