[Bugfix][EP+DP] Fix internode check (#19112)

Signed-off-by: Tyler Michael Smith <tysmith@redhat.com>
This commit is contained in:
Tyler Michael Smith
2025-06-04 11:39:23 -04:00
committed by GitHub
parent c8dcc15921
commit d459fae0a2
2 changed files with 1 additions and 8 deletions

View File

@@ -49,8 +49,7 @@ class All2AllManagerBase:
# all2all communication often has separate implementations for
# intra-node and inter-node communication
self.intranode = in_the_same_node_as(cpu_group, source_rank=0)
self.internode = not self.intranode
self.internode = not all(in_the_same_node_as(cpu_group, source_rank=0))
def get_handle(self, kwargs):
# get a handle for the all2all communication,