forked from mindspore-Ecosystem/mindspore
!12942 Fix UMonad type not supported in Online Debugger
From: @alashkari Reviewed-by: @john_tzanakakis,@mikef Signed-off-by: @john_tzanakakis
This commit is contained in:
commit
b6055f99b2
|
@ -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…
Reference in New Issue