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.

Baizhou Zhang ed4c448488 [pipeline] rewrite t5 tests & support multi-tensor transmitting in pipeline (#4388) ... * fix remaining t5 bugs/rewrite t5 tests * fix multi-tensor communication in pipeline * rearrange test_config * fix keyerror in sync_shared_params * fix get_held_layers & Randomnizer, complete t5 tests * erase printing * fix get_held_layers through modifying _release_unheld_layers * fix _get_recursive_held_layers bug		1 year ago
..
_C	…
_analyzer	…
amp	[pipeline] support fp32 for HybridPlugin/merge shardformer test and pipeline test into one file (#4354)	1 year ago
auto_parallel	…
autochunk	…
booster	[pipeline] rewrite t5 tests & support multi-tensor transmitting in pipeline (#4388)	1 year ago
builder	…
checkpoint_io	…
cli	…
cluster	[cluster] add process group mesh (#4039)	1 year ago
communication	…
context	…
device	…
engine	…
fx	…
interface	[pipeline] refactor 1f1b schedule (#4115)	1 year ago
kernel	…
lazy	[shardformer] support lazy init (#4202)	1 year ago
logging	…
nn	…
pipeline	[pipeline] rewrite t5 tests & support multi-tensor transmitting in pipeline (#4388)	1 year ago
registry	…
shardformer	[pipeline] rewrite t5 tests & support multi-tensor transmitting in pipeline (#4388)	1 year ago
tensor	[pipeline] support fp32 for HybridPlugin/merge shardformer test and pipeline test into one file (#4354)	1 year ago
testing	…
trainer	…
utils	…
zero	…
__init__.py	…
constants.py	…
core.py	…
global_variables.py	…
initialize.py	…