p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add testcase related to auto-monad

This commit is contained in:
huangbingjian 2021-02-22 16:14:07 +08:00
parent cab5ece4e9
commit a158b2f2fa
1 changed files with 40 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
tests/st/auto_monad/test_auto_monad.py
View File

@ -1467,3 +1467,43 @@ def test_multi_add_assign():
outputs = [r2.asnumpy(), r1.asnumpy(), net.p.data.asnumpy()]
expects = numpy_out(p, i0, i1, i2)
np.testing.assert_array_equal(outputs, expects)
@pytest.mark.skip(reason="not supported yet")
def test_multi_abs_add_assign():
class Net(Cell):
def __init__(self, para):
super(Net, self).__init__()
self.add = P.Add()
self.sub = P.Sub()
self.mul = P.Mul()
self.abs = P.Abs()
self.assign = P.Assign()
self.p = Parameter(para, name='para')
def construct(self, a, d, e):
tmp = self.abs(self.add(self.abs(a), self.abs(self.mul(a, a))))
res1 = self.add(self.p, tmp)
mul = self.mul(d, e)
self.assign(self.p, mul)
res2 = self.sub(self.p, e)
return res2, res1, tmp
def numpy_out(p, a, d, e):
tmp = np.abs(np.abs(a) + np.abs(a * a))
res1 = p + tmp
res_as = d * e
res2 = d * e - e
return res2, res1, res_as, tmp
p = -(np.abs(np.random.normal(0, 1, [3])) + 1).astype(np.float32)
i0 = -(np.abs(np.random.normal(0, 1, [3])) + 1).astype(np.float32)
i1 = -(np.abs(np.random.normal(0, 1, [3])) + 1).astype(np.float32)
i2 = -(np.abs(np.random.normal(0, 1, [3])) + 1).astype(np.float32)
net = Net(Tensor(p))
r2, r1, tmp = net(Tensor(i0), Tensor(i1), Tensor(i2))
outputs = [r2.asnumpy(), r1.asnumpy(), net.p.data.asnumpy(), tmp.asnumpy()]
expects = numpy_out(p, i0, i1, i2)
np.testing.assert_array_equal(outputs, expects)