ColossalAI/colossalai/core.py

7 lines
145 B
Python

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
from colossalai.context import ParallelContext
global_context = ParallelContext.get_instance()