Browse Source

add global step info for known subgraph in unknown model and generate om for remained cases when some single op cases run atc failed

tags/v1.5.1
lichun 3 years ago
parent
commit
47b3762f6e
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      ge/hybrid/node_executor/compiledsubgraph/known_node_executor.cc
  2. +1
    -1
      tests/ut/ge/hybrid/known_node_executor_unittest.cc

+ 1
- 1
ge/hybrid/node_executor/compiledsubgraph/known_node_executor.cc View File

@@ -190,7 +190,7 @@ void KnownNodeExecutor::SettingDaviciModel(const HybridModel &model, const NodeP
davinci_model->SetDumpModelName(model.GetModelName());
davinci_model->SetOmName(model.GetOmName());
TensorValue *global_step_var = model.GetVariable(NODE_NAME_GLOBAL_STEP);
davinci_model->SetKnownShapeGlobalStep(global_step_var->MutableData());
davinci_model->SetGlobalStep(global_step_var->MutableData());
// set model id as root node's node id
davinci_model->SetSubModelId(node->GetOpDesc()->GetId());
}


+ 1
- 1
tests/ut/ge/hybrid/known_node_executor_unittest.cc View File

@@ -142,5 +142,5 @@ TEST_F(UnknownNodeExecutorTest, TestSetGlobalStep) {
KnownNodeExecutor known_node_executor;
std::shared_ptr<DavinciModel> davinci_model = MakeShared<DavinciModel>(0, nullptr);
known_node_executor.SettingDaviciModel(hybrid_model, node, davinci_model);
EXPECT_EQ(davinci_model->global_step_addr_, 520);
EXPECT_EQ(*(static_cast<int64_t*>(davinci_model->global_step_addr_)), 520);
}

Loading…
Cancel
Save