From 92cb18faabc5fd2d699b02304700c9cbff86947f Mon Sep 17 00:00:00 2001
From: tanghuikang <tanghuikang@huawei.com>
Date: Thu, 7 Jan 2021 11:13:24 +0800
Subject: [PATCH] Fix GetOpInputTensors bug

---
 mindspore/ccsrc/backend/session/ascend_session.cc | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/mindspore/ccsrc/backend/session/ascend_session.cc b/mindspore/ccsrc/backend/session/ascend_session.cc
index eb9c215754..9658c3af48 100644
--- a/mindspore/ccsrc/backend/session/ascend_session.cc
+++ b/mindspore/ccsrc/backend/session/ascend_session.cc
@@ -268,10 +268,10 @@ void GetOpInputTensors(const CNodePtr &cnode, const std::map<KernelWithIndex, te
     MS_EXCEPTION_IF_NULL(real_input);
     tensor::TensorPtr tensor = nullptr;
     if (real_input->isa<ValueNode>()) {
-      auto value_node = input->cast<ValueNodePtr>();
+      auto value_node = real_input->cast<ValueNodePtr>();
       MS_EXCEPTION_IF_NULL(value_node);
       auto value = GetValueNode(value_node);
-      MS_EXCEPTION_IF_NULL(value_node);
+      MS_EXCEPTION_IF_NULL(value);
       if (value->isa<ValueTuple>()) {
         auto value_tuple = value->cast<ValueTuplePtr>();
         MS_EXCEPTION_IF_NULL(value_tuple);
@@ -881,7 +881,7 @@ void AscendSession::RunOpImpl(const GraphInfo &graph_info, OpRunInfo *op_run_inf
 
 void AscendSession::RunOpsInGraphImpl(const GraphId &graph_id, const std::vector<tensor::TensorPtr> &inputs,
                                       VectorRef *outputs) {
-  MS_LOG(INFO) << "Start";
+  MS_LOG(INFO) << "Start!";
   auto kernel_graph = GetGraph(graph_id);
   std::map<AnfNodePtr, size_t> parameter_index;
   GetParameterIndex(kernel_graph.get(), inputs, &parameter_index);
@@ -910,6 +910,7 @@ void AscendSession::RunOpsInGraphImpl(const GraphId &graph_id, const std::vector
     HandleOpInputs(input_tensor_info.input_kernel, &cnode_ref, &op_output_map);
     HandleOpOutputs(kernel, op_outputs, output_indexes, cnode_ref, &op_output_map, outputs);
   }
+  MS_LOG(INFO) << "Finish!";
 }
 
 // compile graph steps