From 1511d50accb9e3a38b7eb0e9fefdae82bc79754e Mon Sep 17 00:00:00 2001 From: lichun Date: Sat, 17 Apr 2021 16:44:16 +0800 Subject: [PATCH] mark ATTR_SINGLE_OP_SCENE in offline scene --- ge/generator/ge_generator.cc | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc index a800c415..3e53ea45 100644 --- a/ge/generator/ge_generator.cc +++ b/ge/generator/ge_generator.cc @@ -783,9 +783,7 @@ Status GeGenerator::BuildSingleOp(OpDescPtr &op_desc, const vector &in GELOGD("Inputs size is %zu, outputs size is %zu.", inputs.size(), outputs.size()); GE_CHECK_NOTNULL_EXEC(impl_, return PARAM_INVALID); impl_->is_offline_ = is_offline; - if (!is_offline) { - (void)AttrUtils::SetBool(op_desc, ATTR_SINGLE_OP_SCENE, true); - } + (void)AttrUtils::SetBool(op_desc, ATTR_SINGLE_OP_SCENE, true); if (CheckForSingleOp(op_desc, inputs, outputs) != SUCCESS) { GELOGE(PARAM_INVALID, "input param is invalid when build single op!");