Skip to content

Commit 2e28c79

Browse files
GuangyaoZhangver217
authored andcommitted
[compatibility] support torch 2.2 (#5875)
* Support Pytorch 2.2.2 * keep build_on_pr file and update .compatibility
1 parent d8bf7e0 commit 2e28c79

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

.compatibility

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,2 @@
11
2.1.0-12.1.0
2+
2.2.2-12.1.0

colossalai/tensor/d_tensor/layout_converter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -473,7 +473,7 @@ def _group_alive_check(cached_comm_action_sequence):
473473
for process_group in used_process_groups:
474474
try:
475475
dist.get_rank(process_group)
476-
except RuntimeError as e:
476+
except (ValueError, RuntimeError) as e:
477477
# If the group is not registered, it means it has been deleted
478478
if str(e) == (
479479
f"Group {process_group} is not registered, please create group with torch.distributed.new_group API"

0 commit comments

Comments
 (0)