diff --git a/paddle/fluid/operators/transpose_op.cc b/paddle/fluid/operators/transpose_op.cc index 48024cbb3ca171b7eaef28bae7a103d44fa5d07a..764331c4bc976796cf49a62f62c78615602d1dc6 100644 --- a/paddle/fluid/operators/transpose_op.cc +++ b/paddle/fluid/operators/transpose_op.cc @@ -231,14 +231,6 @@ class Transpose2Op : public TransposeOp { const framework::ExecutionContext &ctx) const override { framework::proto::VarType::Type data_type = OperatorWithKernel::IndicateVarDataType(ctx, "X"); -#ifdef PADDLE_WITH_MKLDNN - if (this->CanMKLDNNBeUsed(ctx, data_type)) { - return framework::OpKernelType(data_type, - ctx.GetPlace(), - framework::DataLayout::kMKLDNN, - framework::LibraryType::kMKLDNN); - } -#endif std::string data_format = ctx.Attr("data_format"); framework::DataLayout layout_ = framework::StringToDataLayout(data_format); return framework::OpKernelType(data_type, ctx.GetPlace(), layout_); diff --git a/paddle/fluid/platform/mkldnn_op_list.h b/paddle/fluid/platform/mkldnn_op_list.h index 76f0a2affcc74208f6310953d14562ede93910a0..376aa35f2bb8d11fc43d416dd25c7a2561f4a901 100644 --- a/paddle/fluid/platform/mkldnn_op_list.h +++ b/paddle/fluid/platform/mkldnn_op_list.h @@ -78,8 +78,7 @@ static const std::unordered_set mkldnn_white_list = { "prior_box", "fc", "mul", - "mul_grad", - "transpose2"}; + "mul_grad"}; inline bool in_mkldnn_white_list(const std::string& op_name) { return mkldnn_white_list.find(op_name) != mkldnn_white_list.end();