minor formatting change

PaddlePaddle · Apr 14, 2021 · ffe6156 · ffe6156
1 parent bd69270
commit ffe6156
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 59 deletions.
diff --git a/paddle/fluid/operators/reduce_ops/mkldnn/reduce_mkldnn_op.h b/paddle/fluid/operators/reduce_ops/mkldnn/reduce_mkldnn_op.h
@@ -132,7 +132,6 @@ class ReduceGradMKLDNNKernel : public framework::OpKernel<T> {
     const auto& onednn_engine = dev_ctx.GetEngine();
 
     auto dims = ctx.Attr<std::vector<int>>("dim");
-    auto* input_x = ctx.Input<LoDTensor>("X");
     auto* input_dy = ctx.Input<Tensor>(framework::GradVarName("Out"));
 
     auto* output_dx = ctx.Output<Tensor>(framework::GradVarName("X"));
@@ -181,6 +180,7 @@ class ReduceGradMKLDNNKernel : public framework::OpKernel<T> {
       default:
         platform::errors::InvalidArgument(
             "Tensor dims must be in range <1, 5>");
+        return mkldnn:memory:format_tag::a;
     }
   }
 };

diff --git a/python/paddle/fluid/tests/unittests/mkldnn/test_reduce_grad_tmp.py b/python/paddle/fluid/tests/unittests/mkldnn/test_reduce_grad_tmp.py