Skip to content

Commit fa888c1

Browse files
committed
Fix
1 parent e561677 commit fa888c1

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

paddle/fluid/framework/ir/onednn/activation_onednn_fuse_pass.h

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -61,17 +61,19 @@ inline void SetActivationAttrs(paddle::framework::OpDesc* fused_op,
6161
const std::string& act_type) {
6262
bool use_mkldnn = false;
6363
if (fused_op->HasAttr("use_mkldnn") && !fused_op->HasAttr("use_onednn")) {
64-
PADDLE_ENFORCE(PADDLE_GET_CONST(bool, fused_op->GetAttr("use_mkldnn")),
65-
common::errors::PreconditionNotMet(
66-
"oneDNN activation fuses require use_onednn=True"));
64+
PADDLE_ENFORCE_EQ(PADDLE_GET_CONST(bool, fused_op->GetAttr("use_mkldnn")),
65+
true,
66+
common::errors::PreconditionNotMet(
67+
"oneDNN activation fuses require use_onednn=True"));
6768
}
6869
if (fused_op->HasAttr("use_mkldnn")) {
6970
use_mkldnn = PADDLE_GET_CONST(bool, fused_op->GetAttr("use_mkldnn"));
7071
}
7172
if (!use_mkldnn && fused_op->HasAttr("use_onednn")) {
72-
PADDLE_ENFORCE(PADDLE_GET_CONST(bool, fused_op->GetAttr("use_onednn")),
73-
common::errors::PreconditionNotMet(
74-
"oneDNN activation fuses require use_onednn=True"));
73+
PADDLE_ENFORCE_EQ(PADDLE_GET_CONST(bool, fused_op->GetAttr("use_onednn")),
74+
true,
75+
common::errors::PreconditionNotMet(
76+
"oneDNN activation fuses require use_onednn=True"));
7577
}
7678
fused_op->SetAttr("use_onednn", true);
7779

0 commit comments

Comments
 (0)