Zheng Zangwei (Alex Zheng) a6a0325b47 change flashattn to flash_attn (#387)		2024-05-10 14:36:31 +08:00
..
test_attn.py	change flashattn to flash_attn (#387)	2024-05-10 14:36:31 +08:00
test_np_torch.py
test_pos_emb.py
test_seq_parallel_attention.py	change flashattn to flash_attn (#387)	2024-05-10 14:36:31 +08:00
test_t5_shardformer.py