diff --git a/tests/st/dynamic_shape/test_dynamic_shape_with_control_flow.py b/tests/st/dynamic_shape/test_dynamic_shape_with_control_flow.py
index 15cfa4fd0a2..d7ec2183979 100644
--- a/tests/st/dynamic_shape/test_dynamic_shape_with_control_flow.py
+++ b/tests/st/dynamic_shape/test_dynamic_shape_with_control_flow.py
@@ -55,6 +55,8 @@ class UniqueWhile(nn.Cell):
 
 @pytest.mark.level0
 @pytest.mark.platform_x86_gpu_training
+@pytest.mark.platform_arm_ascend_training
+@pytest.mark.platform_x86_ascend_training
 @pytest.mark.env_onecard
 def test_unique_if():
     """
@@ -72,6 +74,8 @@ def test_unique_if():
 
 @pytest.mark.level0
 @pytest.mark.platform_x86_gpu_training
+@pytest.mark.platform_arm_ascend_training
+@pytest.mark.platform_x86_ascend_training
 @pytest.mark.env_onecard
 def test_unique_while():
     """