mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-01-15 23:23:11 +00:00
4f59693207
* [fx] compute memory stat and flop count for MetaInfoProp. * [fx] modify node attribute. * [fx] modify ckpt_chen. * [fx] fix compatibility. * [fx] fix import error. * [fx] skip test for MetaInfoProp. * [fx] skip test for MetaInfoProp. * [fx] skip test for MetaInfoProp. * [fx] skip test for MetaInfoProp. * [fx] skip if torch 1.11.0. * [fx] recover MetaInfoProp support for PyTorch 1.11. * [fx] provide a stable but not accurate enough version of profiler. * [fx] provide a stable but not accurate enough version of profiler. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix import error.
12 lines
394 B
Python
12 lines
394 B
Python
try:
|
|
from . import _meta_registrations
|
|
META_COMPATIBILITY = True
|
|
except:
|
|
import torch
|
|
META_COMPATIBILITY = False
|
|
print(f'_meta_registrations seems to be incompatible with PyTorch {torch.__version__}.')
|
|
from .initialize import (initialize, launch, launch_from_openmpi, launch_from_slurm, launch_from_torch,
|
|
get_default_parser)
|
|
|
|
__version__ = '0.1.9'
|