Skip to content

Commit e64604d

Browse files
committed
Fix
1 parent bb8194e commit e64604d

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

paddle/phi/kernels/fusion/onednn/fc_kernel.cc

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -570,27 +570,28 @@ void FCKernel(const Context& dev_ctx,
570570
const std::string& activation_type,
571571
const bool padding_weights,
572572
DenseTensor* out) {
573-
const bool use_onednn =
573+
const bool use_mkldnn =
574574
dev_ctx.HasDnnAttr("use_mkldnn")
575575
? PADDLE_GET_CONST(bool, dev_ctx.GetDnnAttr("use_mkldnn"))
576576
: false;
577-
use_onednn = dev_ctx.HasDnnAttr("use_onednn")
578-
? PADDLE_GET_CONST(bool, dev_ctx.GetDnnAttr("use_onednn"))
579-
: use_onednn;
577+
const bool use_onednn =
578+
dev_ctx.HasDnnAttr("use_onednn")
579+
? PADDLE_GET_CONST(bool, dev_ctx.GetDnnAttr("use_onednn"))
580+
: use_mkldnn;
580581
const bool use_quantizer =
581582
dev_ctx.HasDnnAttr("use_quantizer")
582583
? PADDLE_GET_CONST(bool, dev_ctx.GetDnnAttr("use_quantizer"))
583584
: false;
584-
const std::string onednn_data_type =
585+
const std::string mkldnn_data_type =
585586
dev_ctx.HasDnnAttr("mkldnn_data_type")
586587
? PADDLE_GET_CONST(std::string,
587588
dev_ctx.GetDnnAttr("mkldnn_data_type"))
588589
: "float32";
589-
onednn_data_type =
590+
const std::string onednn_data_type =
590591
dev_ctx.HasDnnAttr("onednn_data_type")
591592
? PADDLE_GET_CONST(std::string,
592593
dev_ctx.GetDnnAttr("onednn_data_type"))
593-
: onednn_data_type;
594+
: mkldnn_data_type;
594595
const float scale_in =
595596
dev_ctx.HasDnnAttr("Scale_in")
596597
? PADDLE_GET_CONST(float, dev_ctx.GetDnnAttr("Scale_in"))

0 commit comments

Comments
 (0)