From 634eecb98eed79d885144d234654ea4091cabd60 Mon Sep 17 00:00:00 2001 From: ver217 Date: Thu, 23 Jun 2022 11:35:25 +0800 Subject: [PATCH] mark sanity_check of dist_spec_mgr as staticmethod (#1161) --- colossalai/tensor/dist_spec_mgr.py | 1 + 1 file changed, 1 insertion(+) diff --git a/colossalai/tensor/dist_spec_mgr.py b/colossalai/tensor/dist_spec_mgr.py index f76b624c9..9bb91ad3e 100644 --- a/colossalai/tensor/dist_spec_mgr.py +++ b/colossalai/tensor/dist_spec_mgr.py @@ -43,6 +43,7 @@ class DistSpecManager: _use_autograd_function: bool = True + @staticmethod def _sanity_check(old_dist_spec: _DistSpec, dist_spec: _DistSpec) -> None: if old_dist_spec.process_group is not None and old_dist_spec.process_group != dist_spec.process_group \ and dist_spec.process_group is not None: