mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-23 18:20:58 +00:00
554aa9592e
* [legacy] move communication to legacy (#4640) * [legacy] refactor logger and clean up legacy codes (#4654) * [legacy] make logger independent to gpc * [legacy] make optim independent to registry * [legacy] move test engine to legacy * [legacy] move nn to legacy (#4656) * [legacy] move nn to legacy * [checkpointio] fix save hf config * [test] remove useledd rpc pp test * [legacy] fix nn init * [example] skip tutorial hybriad parallel example * [devops] test doc check * [devops] test doc check
10 lines
317 B
Python
10 lines
317 B
Python
from .addmm import colo_addmm
|
|
from .batch_norm import colo_batch_norm
|
|
from .element_wise import *
|
|
from .embedding import colo_embedding
|
|
from .embedding_bag import colo_embedding_bag
|
|
from .layernorm import colo_layernorm
|
|
from .linear import colo_linear
|
|
from .loss import colo_cross_entropy
|
|
from .view import colo_view
|