Boyuan Yao a2b43e393d [autoparallel] Patch meta information of `torch.nn.Embedding` (#2760) ... * [autoparallel] embedding metainfo * [autoparallel] fix function name in test_activation_metainfo * [autoparallel] undo changes in activation metainfo and related tests		2023-02-17 10:39:48 +08:00
..
meta_registry	[autoparallel] Patch meta information of `torch.nn.Embedding` (#2760)	2023-02-17 10:39:48 +08:00
__init__.py	[fx] Add linear metainfo class for auto parallel (#1783)	2022-11-04 10:55:09 +08:00
constants.py	[autoparallel] align the data_ptr with the old version of auto activation checkpoint pipeline (#2261)	2023-01-03 10:30:15 +08:00
metainfo.py	[autoparallel] align the data_ptr with the old version of auto activation checkpoint pipeline (#2261)	2023-01-03 10:30:15 +08:00
registry.py	[fx] Add linear metainfo class for auto parallel (#1783)	2022-11-04 10:55:09 +08:00