From 6c2a120bed8658015f0f4e4ee95cbbe314b6ce5e Mon Sep 17 00:00:00 2001
From: duanjunwen <935724073@qq.com>
Date: Mon, 9 Sep 2024 10:16:03 +0000
Subject: [PATCH] [fix] add testcase with microbatch 4;

---
 .../test_schedule/test_zerobubble_pp.py        | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/tests/test_pipeline/test_schedule/test_zerobubble_pp.py b/tests/test_pipeline/test_schedule/test_zerobubble_pp.py
index 3fbbe6ed0..825c192d8 100644
--- a/tests/test_pipeline/test_schedule/test_zerobubble_pp.py
+++ b/tests/test_pipeline/test_schedule/test_zerobubble_pp.py
@@ -509,15 +509,15 @@ def run_fwd_bwd_iter_input(test_config):
             "precision": "bf16",
             "num_model_chunk": 2,
         },
-        # {
-        #     "batch_size": 8,
-        #     "tp_size": 1,
-        #     "pp_size": 4,
-        #     "num_microbatches": 8,
-        #     "zero_stage": 1,
-        #     "precision": "bf16",
-        #     "num_model_chunk": 2,
-        # },
+        {
+            "batch_size": 8,
+            "tp_size": 1,
+            "pp_size": 4,
+            "num_microbatches": 8,
+            "zero_stage": 1,
+            "precision": "bf16",
+            "num_model_chunk": 2,
+        },
     ],
 )
 def run_fwd_bwd_vschedule_with_optim(test_config):