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
dcdfae9453
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/load/model_manager/davinci_model.cc

+ 1
- 1
ge/graph/load/model_manager/davinci_model.cc View File

@@ -4372,7 +4372,7 @@ void DavinciModel::SetDataDumperArgs(const ComputeGraphPtr &graph, const map<str
data_dumper_.SetDeviceId(device_id); data_dumper_.SetDeviceId(device_id);


if (known_node_) { if (known_node_) {
data_dumper_.SetLoopAddr(global_step_addr, nullptr, nullptr);
data_dumper_.SetLoopAddr(global_step_addr_, nullptr, nullptr);
} else { } else {
// set loop count addr // set loop count addr
auto get_var_addr = [&](const string &name) -> void *{ auto get_var_addr = [&](const string &name) -> void *{


Loading…
Cancel
Save