You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/auto_parallel/meta_profiler
Boyuan Yao b904748210
[autoparallel] bypass MetaInfo when unavailable and modify BCAST_FUNC_OP metainfo (#2293)
2 years ago
..
meta_registry [autoparallel] bypass MetaInfo when unavailable and modify BCAST_FUNC_OP metainfo (#2293) 2 years ago
__init__.py [fx] Add linear metainfo class for auto parallel (#1783) 2 years ago
constants.py [autoparallel] align the data_ptr with the old version of auto activation checkpoint pipeline (#2261) 2 years ago
metainfo.py [autoparallel] align the data_ptr with the old version of auto activation checkpoint pipeline (#2261) 2 years ago
registry.py [fx] Add linear metainfo class for auto parallel (#1783) 2 years ago