fix error type in ccl.py (microsoft#4521)

Co-authored-by: Logan Adams <[email protected]>
baodii · Nov 7, 2023 · 8aa5dab · 8aa5dab
1 parent 96839f6
commit 8aa5dab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspeed/comm/ccl.py b/deepspeed/comm/ccl.py
@@ -170,7 +170,7 @@ def get_all_ranks_from_group(self, group):
             while True:
                 results.append(super(CCLBackend, self).get_global_rank(group, rank))
                 rank += 1
-        except RuntimeError:
+        except ValueError:
             pass
         if tuple(results) not in self.groups:
             self._new_group(results, group)