diff --git a/tests/st/ops/ascend/test_gru_op.py b/tests/st/ops/ascend/test_gru_op.py index eec0e0be083..1c30dfad8ff 100644 --- a/tests/st/ops/ascend/test_gru_op.py +++ b/tests/st/ops/ascend/test_gru_op.py @@ -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 diff --git a/tests/st/ops/ascend/test_lstm_op.py b/tests/st/ops/ascend/test_lstm_op.py index 6218aa88b77..5b9866c0820 100644 --- a/tests/st/ops/ascend/test_lstm_op.py +++ b/tests/st/ops/ascend/test_lstm_op.py @@ -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