Zhongkai Zhao 5d9a0ae75b [hotfix] Fix ShardFormer test execution path when using sequence parallelism (#5230)

2024-01-17 17:42:29 +08:00

..

test_hybrid_parallel_grad_clip_norm

[gemini] gemini support extra-dp (#5043)

2023-11-16 21:03:04 +08:00

test_layer

[shardformer] llama support DistCrossEntropy (#5176)

2023-12-13 01:39:14 +08:00

test_model

[hotfix] Fix ShardFormer test execution path when using sequence parallelism (#5230)

2024-01-17 17:42:29 +08:00

__init__.py

[shardformer] adapted T5 and LLaMa test to use kit (#4049)

2023-07-04 16:05:01 +08:00

test_shard_utils.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

test_with_torch_ddp.py

[ci] fixed ddp test (#5254)

2024-01-11 17:16:32 +08:00