From 8ba58eb064b26e73e12d0c93f47d6cce0a6ceaae Mon Sep 17 00:00:00 2001
From: zmx <zmxdream@pku.edu.cn>
Date: Fri, 3 Sep 2021 14:08:01 +0800
Subject: [PATCH] add AsExtra to partial_sum op (#35381)

add AsExtra to the following attribute of partial_sum op:
1. use_mkldnn
---
 paddle/fluid/operators/partial_sum_op.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/partial_sum_op.cc b/paddle/fluid/operators/partial_sum_op.cc
index 76e4938368..fb60ab54b7 100644
--- a/paddle/fluid/operators/partial_sum_op.cc
+++ b/paddle/fluid/operators/partial_sum_op.cc
@@ -143,7 +143,8 @@ class PartialSumOpMaker : public framework::OpProtoAndCheckerMaker {
     AddAttr<bool>(
         "use_mkldnn",
         "(bool, default false) Indicates if MKL-DNN kernel will be used")
-        .SetDefault(false);
+        .SetDefault(false)
+        .AsExtra();
     AddAttr<int>("start_index", "The start index of tensor wanted to be added.")
         .SetDefault(0);
     AddAttr<int>("length", "The length of tensor wanted to be added.")
-- 
GitLab