|
@@ -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; |
|
|