fix-test (#5210)

fix-test fix-test
2025-09-01 09:07:51 +00:00 · 2024-01-03 14:26:13 +08:00
parent 7f3400b560
commit 365671be10
3 changed files with 9 additions and 1 deletions
--- a/colossalai/cluster/process_group_mesh.py
+++ b/colossalai/cluster/process_group_mesh.py
@@ -45,7 +45,7 @@ class ProcessGroupMesh:
        self._ranks_to_group: Dict[Tuple[int, ...], ProcessGroup] = {}
        self._group_to_ranks: Dict[ProcessGroup, Tuple[int, ...]] = {}

-    def __del__(self):
+    def destroy_mesh_process_groups(self):
        r"""
        Destructor method for the ProcessGroupMesh class.