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.

Wang Binluo dcd41d0973 Merge pull request #6071 from wangbluo/ring_attention ... [Ring Attention] fix the 2d ring attn when using multiple machine		1 month ago
..
test_hybrid_parallel_grad_clip_norm	…
test_layer	Merge pull request #6071 from wangbluo/ring_attention	1 month ago
test_model	…
__init__.py	…
test_flash_attention.py	…
test_shard_utils.py	…
test_with_torch_ddp.py	…