core.py

#!/usr/bin/env python
# -*- encoding: utf-8 -*-

from colossalai.context import ParallelContext, MoeContext

global_context = ParallelContext.get_instance()
moe_context = MoeContext.get_instance()