From f9677aa64341494eab5e15971633475bd8ce1a23 Mon Sep 17 00:00:00 2001 From: Henry Date: Sat, 27 Apr 2024 14:48:47 +0800 Subject: [PATCH] update tests --- tests/st/nn/test_transformer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/st/nn/test_transformer.py b/tests/st/nn/test_transformer.py index 8b67c475e62f..e028239c6e8d 100644 --- a/tests/st/nn/test_transformer.py +++ b/tests/st/nn/test_transformer.py @@ -303,7 +303,7 @@ def test_transformerdecoder_cpu(dtype, training, jit): assert result.shape == tgt.shape -@pytest.mark.level1 +@pytest.mark.level0 @pytest.mark.platform_arm_ascend_training @pytest.mark.platform_x86_ascend_training @pytest.mark.env_onecard @@ -401,7 +401,7 @@ def test_transformer_gpu(dtype, training, jit): assert result.shape == tgt.shape -@pytest.mark.level1 +@pytest.mark.level0 @pytest.mark.platform_arm_ascend_training @pytest.mark.platform_x86_ascend_training @pytest.mark.env_onecard -- Gitee