diff --git a/mindspore/ccsrc/device/ascend/dump/data_dumper.cc b/mindspore/ccsrc/device/ascend/dump/data_dumper.cc
index 57ac0e0947d1a94a6a03f26360323c5b4f834375..14f2c2a524d8235eb6f15badb67c36f10fb22507 100644
--- a/mindspore/ccsrc/device/ascend/dump/data_dumper.cc
+++ b/mindspore/ccsrc/device/ascend/dump/data_dumper.cc
@@ -126,8 +126,8 @@ bool DataDumper::KernelNeedDump(const CNodePtr &kernel) const {
     return false;
   }
   MS_EXCEPTION_IF_NULL(kernel);
-  const auto &kernel_set = DataDumpParser::GetInstance().kernel_set();
-  return kernel_set.find(kernel->fullname_with_scope()) != kernel_set.end();
+  // dump all kernel if mode is set 0 in data_dump.json
+  return DataDumpParser::GetInstance().NeedDump(kernel->fullname_with_scope());
 }
 
 void DataDumper::UnloadDumpInfo() {
diff --git a/mindspore/ccsrc/session/session_basic.cc b/mindspore/ccsrc/session/session_basic.cc
index 9e437673c92c40ef3a054b4cc511ec58008d098b..59cc0dd020b75d69bea882849ddb79078904e7b5 100644
--- a/mindspore/ccsrc/session/session_basic.cc
+++ b/mindspore/ccsrc/session/session_basic.cc
@@ -175,8 +175,8 @@ size_t LoadCtrlInputTensor(const std::shared_ptr<KernelGraph> &graph, std::vecto
   if (inputs_params == nullptr) {
     return 0;
   }
-  if (inputs_params->empty()) {
-    MS_LOG(EXCEPTION) << "Illegal empty inputs_params";
+  if (inputs_params->size() < 2) {
+    MS_LOG(EXCEPTION) << "Illegal inputs_params size";
   }
   auto tensor = (*inputs_params)[0];
   MS_EXCEPTION_IF_NULL(tensor);