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
History
Boyuan Yao 05ce3d369f
[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 10:55:09 +08:00
..
meta_registry [fx] Add linear metainfo class for auto parallel (#1783) 2022-11-04 10:55:09 +08:00
__init__.py [fx] Add linear metainfo class for auto parallel (#1783) 2022-11-04 10:55:09 +08:00
metainfo.py [fx] Add linear metainfo class for auto parallel (#1783) 2022-11-04 10:55:09 +08:00
registry.py [fx] Add linear metainfo class for auto parallel (#1783) 2022-11-04 10:55:09 +08:00