Browse Source

!970 infershape parellel process when unknown shape

From: @wan_xuelei
Reviewed-by: @xchu42,@wqtshg
Signed-off-by: @wqtshg
tags/v1.2.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
542e604229
1 changed files with 0 additions and 1 deletions
  1. +0
    -1
      ge/hybrid/executor/worker/shape_inference_engine.cc

+ 0
- 1
ge/hybrid/executor/worker/shape_inference_engine.cc View File

@@ -68,7 +68,6 @@ Status ShapeInferenceEngine::InferShape(NodeState &node_state) {
// Do shape inference // Do shape inference
GELOGD("[%s] Start to invoke InferShapeAndType", node_item.NodeName().c_str()); GELOGD("[%s] Start to invoke InferShapeAndType", node_item.NodeName().c_str());
{ {
std::lock_guard<std::mutex> lk(mu_);
RECORD_SHAPE_INFERENCE_EVENT(execution_context_, node_item.NodeName().c_str(), "[InferShapeAndType] Start"); RECORD_SHAPE_INFERENCE_EVENT(execution_context_, node_item.NodeName().c_str(), "[InferShapeAndType] Start");
GE_CHK_STATUS_RET(ShapeRefiner::InferShapeAndTypeForRunning(node_item.node, true), GE_CHK_STATUS_RET(ShapeRefiner::InferShapeAndTypeForRunning(node_item.node, true),
"Invoke InferShapeAndType failed."); "Invoke InferShapeAndType failed.");


Loading…
Cancel
Save