From 8aa5dab58b2c005b030f41cdc37a2a70dce9be05 Mon Sep 17 00:00:00 2001 From: Liangliang-Ma Date: Thu, 19 Oct 2023 01:04:57 +0800 Subject: [PATCH] fix error type in ccl.py (#4521) Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com> --- deepspeed/comm/ccl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepspeed/comm/ccl.py b/deepspeed/comm/ccl.py index 859e70c7db3b..6e915d0ca430 100644 --- 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)