p9z8vbsoh
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

CI

This commit is contained in:
Henry 2022-03-26 11:29:18 +08:00
parent 7e09ae33cd
commit 8ed5d91c8a
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/st/ops/ascend/test_gru_op.py
View File

@ -88,7 +88,7 @@ class GRUWeightBias():
return w_ih_list, w_hh_list, b_ih_list, b_hh_list
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.env_onecard
@ -129,7 +129,8 @@ def test_sit_gru_forward_input_3_32_32_is_32_hs_16():
assert np.allclose(out.asnumpy(), out_pynative.asnumpy(), 0.001, 0.001)
assert np.allclose(hy.asnumpy(), hy_pynative.asnumpy(), 0.001, 0.001)
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.env_onecard

6
tests/st/ops/ascend/test_lstm_op.py
View File

@ -87,7 +87,8 @@ class LSTMWeightBias():
b_hh_list = ParameterTuple(b_hh_list)
return w_ih_list, w_hh_list, b_ih_list, b_hh_list
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.env_onecard
@ -131,7 +132,8 @@ def test_sit_lstm_forward_input_3_32_32_is_32_hs_16():
assert np.allclose(hy.asnumpy(), hy_pynative.asnumpy(), 0.002, 0.002)
assert np.allclose(cy.asnumpy(), cy_pynative.asnumpy(), 0.002, 0.002)
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.env_onecard