Frank Lee a887f54711 added sp for stdit3 (#131)		2024-06-14 11:10:43 +08:00
..
test_attn.py	[wip]	2024-05-07 14:51:11 +08:00
test_lr_scheduler.py	added sp for stdit3 (#131)	2024-06-14 11:10:43 +08:00
test_np_torch.py
test_pos_emb.py
test_seq_parallel_attention.py	[wip]	2024-05-07 14:51:11 +08:00
test_stdit3_sequence_parallelism.py	added sp for stdit3 (#131)	2024-06-14 11:10:43 +08:00
test_t5_shardformer.py