mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-01-15 23:23:11 +00:00
b5f9e37c70
* [legacy] remove outdated codes of pipeline (#4692) * [legacy] remove cli of benchmark and update optim (#4690) * [legacy] remove cli of benchmark and update optim * [doc] fix cli doc test * [legacy] fix engine clip grad norm * [legacy] remove outdated colo tensor (#4694) * [legacy] remove outdated colo tensor * [test] fix test import * [legacy] move outdated zero to legacy (#4696) * [legacy] clean up utils (#4700) * [legacy] clean up utils * [example] update examples * [legacy] clean up amp * [legacy] fix amp module * [legacy] clean up gpc (#4742) * [legacy] clean up context * [legacy] clean core, constants and global vars * [legacy] refactor initialize * [example] fix examples ci * [example] fix examples ci * [legacy] fix tests * [example] fix gpt example * [example] fix examples ci * [devops] fix ci installation * [example] fix examples ci
13 lines
571 B
Python
13 lines
571 B
Python
from .initialize import launch, launch_from_openmpi, launch_from_slurm, launch_from_torch
|
|
|
|
try:
|
|
# .version will be created by setup.py
|
|
from .version import __version__
|
|
except ModuleNotFoundError:
|
|
# this will only happen if the user did not run `pip install`
|
|
# and directly set PYTHONPATH to use Colossal-AI which is a bad practice
|
|
__version__ = '0.0.0'
|
|
print('please install Colossal-AI from https://www.colossalai.org/download or from source')
|
|
|
|
__all__ = ['launch', 'launch_from_openmpi', 'launch_from_slurm', 'launch_from_torch', '__version__']
|