p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add dynamic st test

This commit is contained in:
hwjiaorui 2021-02-07 17:02:25 +08:00
parent 028dcf85ae
commit 3981b6da0c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/st/dynamic_shape/test_ftrl.py
View File

@ -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