diff --git a/python/paddle/fluid/tests/unittests/test_elementwise_mul_op.py b/python/paddle/fluid/tests/unittests/test_elementwise_mul_op.py index 7a79de512996c16970085d2fc9f180bd2fdb5e9f..97ae132daa537f67e9a837be1ecbd2d66db2eb95 100644 --- a/python/paddle/fluid/tests/unittests/test_elementwise_mul_op.py +++ b/python/paddle/fluid/tests/unittests/test_elementwise_mul_op.py @@ -190,6 +190,7 @@ class TestElementwiseMulOp_commonuse_1(ElementwiseMulOp): 'Y': np.random.rand(1, 1, 4).astype(np.float64) } self.outputs = {'Out': self.inputs['X'] * self.inputs['Y']} + self.init_kernel_type() class TestElementwiseMulOp_commonuse_2(ElementwiseMulOp): @@ -200,6 +201,7 @@ class TestElementwiseMulOp_commonuse_2(ElementwiseMulOp): 'Y': np.random.rand(2, 1, 4, 1).astype(np.float64) } self.outputs = {'Out': self.inputs['X'] * self.inputs['Y']} + self.init_kernel_type() class TestElementwiseMulOp_xsize_lessthan_ysize(ElementwiseMulOp): @@ -215,6 +217,7 @@ class TestElementwiseMulOp_xsize_lessthan_ysize(ElementwiseMulOp): self.outputs = { 'Out': self.inputs['X'].reshape(1, 1, 4, 5) * self.inputs['Y'] } + self.init_kernel_type() class TestElementwiseMulOpError(OpTest):