|
|
|
@ -84,3 +84,54 @@ def test_mul():
|
|
|
|
|
err = np.ones(shape=exp.shape) * 1.0e-5
|
|
|
|
|
assert np.all(diff < err)
|
|
|
|
|
assert out.shape == exp.shape
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.mark.level0
|
|
|
|
|
@pytest.mark.platform_x86_cpu
|
|
|
|
|
@pytest.mark.env_onecard
|
|
|
|
|
def test_mul_int32():
|
|
|
|
|
x0 = Tensor(np.random.uniform(-2, 2, (2, 3, 4, 4)).astype(np.int32))
|
|
|
|
|
y0 = Tensor(np.random.uniform(-2, 2, (1, 1, 1, 1)).astype(np.int32))
|
|
|
|
|
x1 = Tensor(np.random.uniform(-2, 2, (1, 3, 1, 4)).astype(np.int32))
|
|
|
|
|
y1 = Tensor(np.random.uniform(-2, 2, (2, 3, 4, 4)).astype(np.int32))
|
|
|
|
|
x2 = Tensor(np.random.uniform(-2, 2, (2, 3, 4, 4)).astype(np.int32))
|
|
|
|
|
y2 = Tensor(2, mstype.int32)
|
|
|
|
|
x3 = Tensor(2, mstype.int32)
|
|
|
|
|
y3 = Tensor(2, mstype.int32)
|
|
|
|
|
x4 = Tensor(np.random.uniform(-2, 2, (4)).astype(np.int32))
|
|
|
|
|
y4 = Tensor(np.random.uniform(-2, 2, (4, 4)).astype(np.int32))
|
|
|
|
|
mul = Net()
|
|
|
|
|
out = mul(x0, y0).asnumpy()
|
|
|
|
|
exp = x0.asnumpy() * y0.asnumpy()
|
|
|
|
|
diff = np.abs(out - exp)
|
|
|
|
|
err = np.ones(shape=exp.shape) * 1.0e-5
|
|
|
|
|
assert np.all(diff < err)
|
|
|
|
|
assert out.shape == exp.shape
|
|
|
|
|
|
|
|
|
|
out = mul(x1, y1).asnumpy()
|
|
|
|
|
exp = x1.asnumpy() * y1.asnumpy()
|
|
|
|
|
diff = np.abs(out - exp)
|
|
|
|
|
err = np.ones(shape=exp.shape) * 1.0e-5
|
|
|
|
|
assert np.all(diff < err)
|
|
|
|
|
assert out.shape == exp.shape
|
|
|
|
|
|
|
|
|
|
out = mul(x2, y2).asnumpy()
|
|
|
|
|
exp = x2.asnumpy() * y2.asnumpy()
|
|
|
|
|
diff = np.abs(out - exp)
|
|
|
|
|
err = np.ones(shape=exp.shape) * 1.0e-5
|
|
|
|
|
assert np.all(diff < err)
|
|
|
|
|
assert out.shape == exp.shape
|
|
|
|
|
|
|
|
|
|
out = mul(x3, y3).asnumpy()
|
|
|
|
|
exp = x3.asnumpy() * y3.asnumpy()
|
|
|
|
|
diff = np.abs(out - exp)
|
|
|
|
|
err = np.ones(shape=exp.shape) * 1.0e-5
|
|
|
|
|
assert np.all(diff < err)
|
|
|
|
|
assert out.shape == exp.shape
|
|
|
|
|
|
|
|
|
|
out = mul(x4, y4).asnumpy()
|
|
|
|
|
exp = x4.asnumpy() * y4.asnumpy()
|
|
|
|
|
diff = np.abs(out - exp)
|
|
|
|
|
err = np.ones(shape=exp.shape) * 1.0e-5
|
|
|
|
|
assert np.all(diff < err)
|
|
|
|
|
assert out.shape == exp.shape
|
|
|
|
|