From aea6b0c97478cf8268f1edd8a06b2f65c246aec0 Mon Sep 17 00:00:00 2001 From: VectorSL Date: Wed, 1 Apr 2020 11:49:37 +0800 Subject: [PATCH] update tests/st/ops/gpu/test_tensoradd.py. fix pytest.mark for testcase --- tests/st/ops/gpu/test_tensoradd.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/tests/st/ops/gpu/test_tensoradd.py b/tests/st/ops/gpu/test_tensoradd.py index a60f5dcf30e..23c23f12d20 100644 --- a/tests/st/ops/gpu/test_tensoradd.py +++ b/tests/st/ops/gpu/test_tensoradd.py @@ -25,10 +25,6 @@ from mindspore.common.parameter import Parameter context.set_context(device_target='GPU') - -@pytest.mark.level0 -@pytest.mark.platform_x86_gpu_training -@pytest.mark.env_onecard class TensroAdd(nn.Cell): def __init__(self): super(TensroAdd, self).__init__() @@ -61,7 +57,9 @@ class TensroAdd(nn.Cell): self.add(self.x, self.y), self.add(self.x1, self.y1), self.add(self.x2, self.y2), self.add(self.x3, self.y3)) - +@pytest.mark.level0 +@pytest.mark.platform_x86_gpu_training +@pytest.mark.env_onecard def test_TensroAdd(): add = TensroAdd() output = add()