wangtao
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

update tests/st/ops/gpu/test_tensoradd.py. 

fix pytest.mark for testcase
This commit is contained in:
VectorSL 2020-04-01 11:49:37 +08:00 committed by Gitee
parent 569e5b056e
commit aea6b0c974
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/st/ops/gpu/test_tensoradd.py
View File

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