From 3981b6da0c263b2268f843275c96d1c43aa17b5e Mon Sep 17 00:00:00 2001 From: hwjiaorui Date: Sun, 7 Feb 2021 17:02:25 +0800 Subject: [PATCH] add dynamic st test --- tests/st/dynamic_shape/test_ftrl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/st/dynamic_shape/test_ftrl.py b/tests/st/dynamic_shape/test_ftrl.py index 9abecac53ef..050d7474138 100644 --- 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