Hongxin Liu bdb9b7ccbf [hotfix] fix attn impl (#17)		2024-04-01 20:37:15 +08:00
..
test_attn.py	[hotfix] fix attn impl (#17)	2024-04-01 20:37:15 +08:00
test_np_torch.py	replaced numpy ops with torch ops (#9)	2024-03-30 13:01:58 +08:00
test_pos_emb.py	[feature] impl cached pos embedding (#1)	2024-03-28 17:30:47 +08:00
test_seq_parallel_attention.py	format and some fix (#8)	2024-03-30 13:34:19 +08:00
test_t5_shardformer.py