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