update ut

Author: FeixLiu

test/auto_parallel/hybrid_strategy/test_semi_auto_parallel_llama_model.py

@@ -68,7 +68,7 @@ def test_simple_net_hybrid_strategy(self):
 
 class TestSemiAutoParallelLlama3D(test_base.CommunicationTestDistBase):
     def setUp(self):
-        super().setUp(num_of_devices=8, timeout=200, nnode=1)
+        super().setUp(num_of_devices=8, timeout=400, nnode=1)
         self._default_envs = {"dp": "2", "mp": "2", "pp": "2", "acc_step": "2"}
         self._changeable_envs = {
             "backend": ["gpu"],
@@ -91,7 +91,7 @@ def test_simple_net_hybrid_strategy(self):
 
 class TestSemiAutoParallelLlamaACC(test_base.CommunicationTestDistBase):
     def setUp(self):
-        super().setUp(num_of_devices=8, timeout=200, nnode=1)
+        super().setUp(num_of_devices=8, timeout=400, nnode=1)
         self._default_envs = {
             "dp": "2",
             "mp": "2",