mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-10-20 06:09:05 +00:00
05ce3d369f
* [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
4 lines
89 B
Python
4 lines
89 B
Python
from .meta_registry import *
|
|
from .metainfo import *
|
|
from .registry import meta_register
|