Browse Source

Fix UMonad type in online debugger

tags/v1.2.0-rc1
Amir Lashkari 5 years ago
parent
commit
497e717081
1 changed files with 6 additions and 0 deletions
  1. +6
    -0
      mindspore/ccsrc/debug/debugger/debugger.cc

+ 6
- 0
mindspore/ccsrc/debug/debugger/debugger.cc View File

@@ -1135,6 +1135,9 @@ void Debugger::LoadSingleAnfnode(const AnfNodePtr &anf_node, const size_t output
auto addr = AnfAlgo::GetOutputAddr(anf_node, output_index);
MS_EXCEPTION_IF_NULL(addr);
auto type = AnfAlgo::GetOutputInferDataType(anf_node, output_index);
if (type == kObjectTypeUMonad) {
return;
}
auto format = kOpFormat_DEFAULT;
string tensor_name = node_name + ':' + "0";
ShapeVector int_shapes;
@@ -1197,6 +1200,9 @@ void Debugger::LoadGraphOutputs() {
auto addr = AnfAlgo::GetOutputAddr(node, j);
MS_EXCEPTION_IF_NULL(addr);
auto type = AnfAlgo::GetOutputInferDataType(node, j);
if (type == kObjectTypeUMonad) {
continue;
}
auto format = kOpFormat_DEFAULT;
string tensor_name = kernel_name + ':' + std::to_string(j);
ShapeVector int_shapes;


Loading…
Cancel
Save