Update test_shard_deepseek.py

2025-09-23 18:39:56 +00:00 · 2025-05-23 11:16:36 +08:00
parent 252efa63fc
commit b7df86848c
1 changed files with 1 additions and 0 deletions
--- 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),