mirror of https://github.com/hpcaitech/ColossalAI
17 lines
440 B
Python
17 lines
440 B
Python
#!/usr/bin/env python
|
|
# -*- encoding: utf-8 -*-
|
|
|
|
from colossalai.context import ParallelContext
|
|
|
|
global_context = ParallelContext()
|
|
|
|
|
|
def set_global_context(context: ParallelContext):
|
|
'''Reset global context to be identical to a given :class:ParallelContext.
|
|
|
|
:param context: Parallel context to generate our global parallel context.
|
|
:type context: ParallelContext
|
|
'''
|
|
global global_context
|
|
global_context = context
|