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 22e947f982 [autoparallel] fix runtime apply memory estimation (#2281) ... * [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 * [autoparallel] fix wrong runtime apply calculation * [autoparallel] fix wrong runtime apply calculation * [autoparallel] fix wrong runtime apply calculation		2 years ago
..
_C	[optimizer] add div_scale for optimizers (#2117)	2 years ago
amp	[builder] unified cpu_optim fused_optim inferface (#2190)	2 years ago
auto_parallel	[autockpt] considering parameter and optimizer weights. (#2279)	2 years ago
builder	…
cli	[cli] updated installation cheheck with more inforamtion (#2050)	2 years ago
communication	…
context	[hotfix] Fixing the bug related to ipv6 support	2 years ago
device	[device] update flatten device mesh usage (#2079)	2 years ago
engine	…
fx	[autoparallel] modify comm nodes' memory cost in construct chain (#2263)	2 years ago
gemini	[example] update gpt example for larger model scale (#2211)	2 years ago
kernel	[hotfix] fix error for torch 2.0 (#2243)	2 years ago
logging	[logger] hotfix, missing _FORMAT (#2231)	2 years ago
nn	[zero] fix error for BEiT models (#2169)	2 years ago
pipeline	fix type of num_worker_threads (#2237)	2 years ago
registry	…
tensor	[autoparallel] fix runtime apply memory estimation (#2281)	2 years ago
testing	…
trainer	…
utils	[builder] unified cpu_optim fused_optim inferface (#2190)	2 years ago
zero	[example] add zero1, zero2 example in GPT examples (#2146)	2 years ago
__init__.py	[setup] supported conda-installed torch (#2048)	2 years ago
constants.py	…
core.py	…
global_variables.py	…
initialize.py	…