add dynamic st test

4 years ago · 3981b6da0c
parent 028dcf85ae
commit 3981b6da0c
1 changed files with 2 additions and 2 deletions
--- a/tests/st/dynamic_shape/test_ftrl.py
+++ b/tests/st/dynamic_shape/test_ftrl.py
@ -35,7 +35,7 @@ class NetWithSparseGatherV2(nn.Cell):
    def construct(self, indices, label):
        return self.gather(self.weight1, indices, self.axis) + self.weight2

-# @pytest.mark.level0
+@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.env_onecard
@ -56,7 +56,7 @@ def test_ftrl_net():
                                                 [[0.6821311, 0.6821311]],
                                                 [[0.6821311, 0.6821311]]]))

-# @pytest.mark.level0
+@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.env_onecard