diff --git a/tests/test_shardformer/test_model/test_shard_deepseek.py b/tests/test_shardformer/test_model/test_shard_deepseek.py index e0589d3cc..99409074c 100644 --- a/tests/test_shardformer/test_model/test_shard_deepseek.py +++ b/tests/test_shardformer/test_model/test_shard_deepseek.py @@ -195,6 +195,7 @@ def run_deepseek_test(config: Tuple[int, ...]): (0, 1, 4, 2, 1), (0, 1, 1, 4, 1), # (0, 1, 4, 1, 1), # todo: failed pass, need to be fixed + (0, 1, 2, 1, 1), # zero 1: (1, 2, 1, 1, 2), (1, 2, 1, 4, 1),