diff --git a/python/paddle/fluid/tests/unittests/mkldnn/test_conv2d_transpose_mkldnn_op.py b/python/paddle/fluid/tests/unittests/mkldnn/test_conv2d_transpose_mkldnn_op.py index a9b26b86c23b787d79d863bc4cc82bc20c0dec75..b98610760e6c72142f20487dd6d9b0adb7dff7fc 100644 --- a/python/paddle/fluid/tests/unittests/mkldnn/test_conv2d_transpose_mkldnn_op.py +++ b/python/paddle/fluid/tests/unittests/mkldnn/test_conv2d_transpose_mkldnn_op.py @@ -60,8 +60,10 @@ class TestConv2dTransposeMKLDNNOp(TestConv2dTransposeOp): f_c = self.input_size[1] self.filter_size = [f_c, 6, 3, 3] self.groups = 1 + self.dtype = np.float32 def setUp(self): + TestConv2dTransposeOp.setUp(self) output = self.outputs['Output'] diff --git a/python/paddle/fluid/tests/unittests/test_conv2d_transpose_op.py b/python/paddle/fluid/tests/unittests/test_conv2d_transpose_op.py index 02b8aacaf2292b19884ff552c0becd09bd83ea29..607e97ef430e15201197eefc711918a6afd20794 100644 --- a/python/paddle/fluid/tests/unittests/test_conv2d_transpose_op.py +++ b/python/paddle/fluid/tests/unittests/test_conv2d_transpose_op.py @@ -78,7 +78,7 @@ def conv2dtranspose_forward_naive(input_, filter_, attrs): out_h = output_size[0] + pad_h_0 + pad_h_1 out_w = output_size[1] + pad_w_0 + pad_w_1 - out = np.zeros((in_n, out_c, out_h, out_w)) + out = np.zeros((in_n, out_c, out_h, out_w), dtype=input_.dtype) for n in range(in_n): for i in range(in_h): @@ -108,7 +108,7 @@ def conv2dtranspose_forward_naive(input_, filter_, attrs): class TestConv2dTransposeOp(OpTest): def setUp(self): # init as conv transpose - self.dtype = np.float32 + self.dtype = np.float64 self.is_test = False self.use_cudnn = False self.use_mkldnn = False @@ -119,8 +119,8 @@ class TestConv2dTransposeOp(OpTest): self.init_op_type() self.init_test_case() - input_ = np.random.random(self.input_size).astype("float32") - filter_ = np.random.random(self.filter_size).astype("float32") + input_ = np.random.random(self.input_size).astype(self.dtype) + filter_ = np.random.random(self.filter_size).astype(self.dtype) self.inputs = {'Input': input_, 'Filter': filter_} self.attrs = { @@ -138,7 +138,7 @@ class TestConv2dTransposeOp(OpTest): self.attrs['output_size'] = self.output_size output = conv2dtranspose_forward_naive(input_, filter_, - self.attrs).astype('float32') + self.attrs).astype(self.dtype) self.outputs = {'Output': output}