Browse Source

Fix bug.

tags/v1.3.0
zhaozhixuan 3 years ago
parent
commit
4bc0f6f2af
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      ge/hybrid/executor/node_state.h
  2. +1
    -1
      ge/hybrid/executor/worker/shape_inference_engine.cc

+ 1
- 1
ge/hybrid/executor/node_state.h View File

@@ -179,7 +179,7 @@ struct NodeState {


void SetSkipInferShape(bool skip_infershape) { skip_infershape_ = skip_infershape; } void SetSkipInferShape(bool skip_infershape) { skip_infershape_ = skip_infershape; }


bool SkipInferShape() const { return skip_infershape_; }
bool MaySkipShapeInference() const { return skip_infershape_; }


private: private:
bool IsScheduleReady() const; bool IsScheduleReady() const;


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

@@ -70,7 +70,7 @@ Status ShapeInferenceEngine::InferShape(NodeState &node_state) {
// Do shape inference // Do shape inference
// Skipping infer shape of input node. // Skipping infer shape of input node.
GELOGD("[%s] Start to invoke InferShapeAndType", node_item.NodeName().c_str()); GELOGD("[%s] Start to invoke InferShapeAndType", node_item.NodeName().c_str());
if (!node_state.SkipInferShape()) {
if (!node_state.MaySkipShapeInference()) {
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] for %s failed.", node_item.NodeName().c_str()); "[Invoke][InferShapeAndType] for %s failed.", node_item.NodeName().c_str());


Loading…
Cancel
Save