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 4b40fbd743
[autoparallel] fix forward memory calculation (#2062) 		2022-12-04 15:00:16 +08:00
..
meta_registry [autoparallel] fix forward memory calculation (#2062) 2022-12-04 15:00:16 +08:00
__init__.py [fx] Add linear metainfo class for auto parallel (#1783) 2022-11-04 10:55:09 +08:00
constants.py [autoparallel] add torch.nn.ReLU metainfo (#1868) 2022-11-16 23:12:31 +08:00
metainfo.py [autoparallel] Add metainfo support for F.linear (#1987) 2022-11-23 14:12:34 +08:00
registry.py [fx] Add linear metainfo class for auto parallel (#1783) 2022-11-04 10:55:09 +08:00