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.

Boyuan Yao 5c2ef9fc76 [autoparallel] modify comm nodes' memory cost in construct chain (#2263) ... * [autoparallel] align the data_ptr with the old version of auto activation checkpoint pipeline * [autoparallel] using fwd_time and bwd_time instead of fwd_flop and bwd_flop * [autoparallel] specifycomm nodes' memory cost in construct chain		2 years ago
..
__init__.py	[autoparallel] refactor the runtime apply pass and add docstring to passes (#1757)	2 years ago
comm_metainfo_pass.py	[autockpt] make it work. (#2257)	2 years ago
constants.py	[autoparallel] align the data_ptr with the old version of auto activation checkpoint pipeline (#2261)	2 years ago
meta_info_prop.py	[autoparallel] modify comm nodes' memory cost in construct chain (#2263)	2 years ago
runtime_apply_pass.py	[autoparallel] Hook all meta information on ResNet nodes for auto activation checkpoint (#2248)	2 years ago
runtime_preparation_pass.py	[autoparallel] new metainfoprop based on metainfo class (#2179)	2 years ago