github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/auto_parallel/meta_profiler/__init__.py

4 lines
95 B
Python
Raw Normal View History

[fx] Add linear metainfo class for auto parallel (#1783) * [fx] metainfo class for auto parallel * [fx] add unit test for linear metainfo * [fx] fix bwd param for linear * [fx] modify unit test * [fx] modify unit test * [fx] modify import * [fx] modify import * [fx] modify import * [fx] move meta profiler to auto parallel
2022-11-04 02:55:09 +00:00
from .meta_registry import *
from .registry import meta_register
[autoparallel]integrate auto parallel feature with new tracer (#3408) * [autoparallel] integrate new analyzer in module level * unify the profiling method * polish * fix no codegen bug * fix pass bug * fix liveness test * polish
2023-04-04 09:40:45 +00:00
from .shard_metainfo import *