Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "9edeadfb24bd769b33d500e05784d63bf1a8e5fd"
Unverified Commit 634eecb9 authored by ver217's avatar ver217 Committed by GitHub
Browse files

mark sanity_check of dist_spec_mgr as staticmethod (#1161)

parent 955ac912
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
colossalai/tensor/dist_spec_mgr.py
View file @ 634eecb9
...@@ -43,6 +43,7 @@ class DistSpecManager: ...@@ -43,6 +43,7 @@ class DistSpecManager:
_use_autograd_function: bool = True _use_autograd_function: bool = True
@staticmethod
def _sanity_check(old_dist_spec: _DistSpec, dist_spec: _DistSpec) -> None: 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 \ 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: and dist_spec.process_group is not None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment