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()