From 4fa62a265255fb787342b8d1c04555b0a0bb37c6 Mon Sep 17 00:00:00 2001 From: wozna Date: Wed, 10 Feb 2021 12:30:37 +0100 Subject: [PATCH] Remove errors related to paddle enforce --- paddle/fluid/operators/mkldnn/conv_transpose_mkldnn_op.cc | 3 ++- .../unittests/mkldnn/test_conv2d_transpose_bf16_mkldnn_op.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/operators/mkldnn/conv_transpose_mkldnn_op.cc b/paddle/fluid/operators/mkldnn/conv_transpose_mkldnn_op.cc index 49c4c054baec8..e54c0b4b65d9d 100644 --- a/paddle/fluid/operators/mkldnn/conv_transpose_mkldnn_op.cc +++ b/paddle/fluid/operators/mkldnn/conv_transpose_mkldnn_op.cc @@ -66,7 +66,8 @@ class ConvTransposeMKLDNNHandlerT "Got wrong layout = %d for Input tensor.", input->layout())); PADDLE_ENFORCE_NE(input->format(), MKLDNNMemoryFormat::undef, platform::errors::InvalidArgument( - "Got wrong format for Input tensor.")); + "Got wrong format for Input tensor. The input " + "format is undefined.")); PADDLE_ENFORCE_EQ( filter->layout(), DataLayout::kMKLDNN, diff --git a/python/paddle/fluid/tests/unittests/mkldnn/test_conv2d_transpose_bf16_mkldnn_op.py b/python/paddle/fluid/tests/unittests/mkldnn/test_conv2d_transpose_bf16_mkldnn_op.py index d672ddccdc376..c6b7c175d9000 100644 --- a/python/paddle/fluid/tests/unittests/mkldnn/test_conv2d_transpose_bf16_mkldnn_op.py +++ b/python/paddle/fluid/tests/unittests/mkldnn/test_conv2d_transpose_bf16_mkldnn_op.py @@ -35,7 +35,7 @@ def conv2d_bias_naive(out, bias): "place does not support BF16 evaluation") class TestConv2DTransposeBF16MKLDNNOp(OpTest): def test_check_output(self): - self.check_output(check_dygraph=(self.use_mkldnn == False)) + self.check_output_with_place(core.CPUPlace()) def test_check_grad(self): pass