From da7c73f84715885268366420fbd3e617fe00bc02 Mon Sep 17 00:00:00 2001 From: Adam <38704900+grygielski@users.noreply.github.com> Date: Thu, 2 Apr 2020 04:16:11 +0200 Subject: [PATCH] Delete is_test attribute from activation operators (#23318) * Delete is_test from activation operators test=develop * Revent unneeded changes test=develop --- paddle/fluid/operators/activation_op.cc | 13 ------------- paddle/fluid/operators/gelu_op.cc | 4 ---- .../fluid/operators/mkldnn/activation_mkldnn_op.cc | 10 ++-------- paddle/fluid/platform/mkldnn_reuse.h | 8 +++----- 4 files changed, 5 insertions(+), 30 deletions(-) diff --git a/paddle/fluid/operators/activation_op.cc b/paddle/fluid/operators/activation_op.cc index 124470f0c3c..9e329be9cc2 100644 --- a/paddle/fluid/operators/activation_op.cc +++ b/paddle/fluid/operators/activation_op.cc @@ -53,11 +53,6 @@ static constexpr bool CanInplaceAct() { "(bool, default false) Only used in cudnn kernel, need " \ "install cudnn") \ .SetDefault(false); \ - AddAttr( \ - "is_test", \ - "(bool, default false) Set to true for inference only, false " \ - "for training. Some layers may run faster when this is true.") \ - .SetDefault(false); \ AddComment(OP_COMMENT); \ } \ } @@ -361,10 +356,6 @@ class LeakyReluOpMaker : public framework::OpProtoAndCheckerMaker { AddAttr("use_mkldnn", "(bool, default false) Only used in mkldnn kernel") .SetDefault(false); - AddAttr("is_test", - "(bool, default false) Set to true for inference only, false " - "for training. Some layers may run faster when this is true.") - .SetDefault(false); AddComment(R"DOC( LeakyRelu Activation Operator. @@ -592,10 +583,6 @@ class SwishOpMaker : public framework::OpProtoAndCheckerMaker { AddAttr("use_mkldnn", "(bool, default false) Only used in mkldnn kernel") .SetDefault(false); - AddAttr("is_test", - "(bool, default false) Set to true for inference only, false " - "for training. Some layers may run faster when this is true.") - .SetDefault(false); AddComment(R"DOC( Swish Activation Operator. diff --git a/paddle/fluid/operators/gelu_op.cc b/paddle/fluid/operators/gelu_op.cc index 07fa09aceee..76af785f132 100644 --- a/paddle/fluid/operators/gelu_op.cc +++ b/paddle/fluid/operators/gelu_op.cc @@ -93,10 +93,6 @@ class GeluOpMaker : public framework::OpProtoAndCheckerMaker { "(bool, default false) Only used in cudnn kernel, need " "install cudnn") .SetDefault(false); - AddAttr("is_test", - "(bool, default false) Set to true for inference only, false " - "for training. Some layers may run faster when this is true.") - .SetDefault(false); AddComment(R"DOC( Gelu Activation Operator. diff --git a/paddle/fluid/operators/mkldnn/activation_mkldnn_op.cc b/paddle/fluid/operators/mkldnn/activation_mkldnn_op.cc index b68cb325a7b..ec1fd4e5bcd 100644 --- a/paddle/fluid/operators/mkldnn/activation_mkldnn_op.cc +++ b/paddle/fluid/operators/mkldnn/activation_mkldnn_op.cc @@ -54,10 +54,6 @@ class MKLDNNActivationGradKernel PADDLE_ENFORCE_NE(diff_y->format(), MKLDNNMemoryFormat::undef, "Wrong format set for Input OutGrad tensor"); - PADDLE_ENFORCE_EQ( - ctx.Attr("is_test"), false, - "is_test attribute should be set to False in training phase."); - Functor functor; functor(ctx); } @@ -89,11 +85,9 @@ void eltwise_forward(const framework::ExecutionContext &ctx, auto src_format = src_tz.size() == 2 ? MKLDNNMemoryFormat::nc : x->format(); - bool is_test = ctx.Attr("is_test"); - platform::ActivationMKLDNNHandler handler( - src_tz, algorithm, alpha, beta, src_format, is_test, dev_ctx, - ctx.GetPlace(), ctx.InputName("X")); + src_tz, algorithm, alpha, beta, src_format, dev_ctx, ctx.GetPlace(), + ctx.InputName("X")); auto src_memory_p = handler.AcquireSrcMemory(x); auto dst_memory_p = handler.AcquireDstMemory(y); diff --git a/paddle/fluid/platform/mkldnn_reuse.h b/paddle/fluid/platform/mkldnn_reuse.h index 25b285ccc83..25d50f1fc8a 100644 --- a/paddle/fluid/platform/mkldnn_reuse.h +++ b/paddle/fluid/platform/mkldnn_reuse.h @@ -411,7 +411,7 @@ class ActivationMKLDNNHandler public: ActivationMKLDNNHandler(const std::vector& dims, mkldnn::algorithm algorithm, float alpha, float beta, - const MKLDNNMemoryFormat fmt, bool is_test, + const MKLDNNMemoryFormat fmt, const platform::MKLDNNDeviceContext& dev_ctx, platform::Place cpu_place, const std::string& unique_name) @@ -422,10 +422,8 @@ class ActivationMKLDNNHandler platform::CreateKey(dims, unique_name)) { auto md = mkldnn::memory::desc(dims, platform::MKLDNNGetDataType(), fmt); - this->AcquireForwardPrimitiveDescriptor( - is_test ? mkldnn::prop_kind::forward_inference - : mkldnn::prop_kind::forward_training, - algorithm, md, alpha, beta); + this->AcquireForwardPrimitiveDescriptor(mkldnn::prop_kind::forward_training, + algorithm, md, alpha, beta); } ActivationMKLDNNHandler(const std::vector& dims, -- GitLab