fix unit test bug

PaddlePaddle · Jan 21, 2022 · 9020218 · 9020218 · paddle-bot-old · Jan 21, 2022
1 parent 794e60f
commit 9020218
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/fluid/operators/reduce_ops/reduce_op.h b/paddle/fluid/operators/reduce_ops/reduce_op.h
@@ -551,7 +551,7 @@ class ReduceOp : public framework::OperatorWithKernel {
 
   framework::KernelSignature GetExpectedPtenKernelArgs(
       const framework::ExecutionContext& ctx) const override {
-    bool reduce_all = ctx.Attr<int>("reduce_all");
+    bool reduce_all = ctx.Attr<bool>("reduce_all");
     if (Type() == "reduce_sum") {
       if (ctx.InputVar("X")->IsType<framework::LoDTensor>()) {
         if (!reduce_all) {