Browse Source

!12230 add dynamic st test

From: @hwjiaorui
Reviewed-by: 
Signed-off-by:
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
1d640e50c9
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      tests/st/dynamic_shape/test_ftrl.py

+ 2
- 2
tests/st/dynamic_shape/test_ftrl.py View File

@@ -35,7 +35,7 @@ class NetWithSparseGatherV2(nn.Cell):
def construct(self, indices, label): def construct(self, indices, label):
return self.gather(self.weight1, indices, self.axis) + self.weight2 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_arm_ascend_training
@pytest.mark.platform_x86_ascend_training @pytest.mark.platform_x86_ascend_training
@pytest.mark.env_onecard @pytest.mark.env_onecard
@@ -56,7 +56,7 @@ def test_ftrl_net():
[[0.6821311, 0.6821311]], [[0.6821311, 0.6821311]],
[[0.6821311, 0.6821311]]])) [[0.6821311, 0.6821311]]]))


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


Loading…
Cancel
Save