Browse Source

fix

tags/v1.3.0
wjm 3 years ago
parent
commit
ab4118108d
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/hybrid/node_executor/compiledsubgraph/known_node_executor.cc

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

@@ -126,7 +126,7 @@ Status KnownNodeTask::Init(TaskContext &context) {
auto dump_properties = context.GetDumpProperties(); auto dump_properties = context.GetDumpProperties();
if (dump_properties.IsDumpOpen() || dump_properties.IsOpDebugOpen()) { if (dump_properties.IsDumpOpen() || dump_properties.IsOpDebugOpen()) {
davinci_model_->SetDumpProperties(dump_properties); davinci_model_->SetDumpProperties(dump_properties);
void *global_step = context.GetExecutionContext()->global_step;;
void *global_step = context.GetExecutionContext()->global_step;
davinci_model_->SetKnownShapeGlobalStep(global_step); davinci_model_->SetKnownShapeGlobalStep(global_step);
} }
int32_t device_id = 0; int32_t device_id = 0;


Loading…
Cancel
Save