| @@ -40,9 +40,7 @@ void CheckIfValidType(const TypePtr &type, debugger::TypeProto *type_proto) { | |||||
| type->isa<UMonadType>() || type->isa<IOMonadType>())) { | type->isa<UMonadType>() || type->isa<IOMonadType>())) { | ||||
| MS_LOG(EXCEPTION) << "Unknown type: " << type->type_name(); | MS_LOG(EXCEPTION) << "Unknown type: " << type->type_name(); | ||||
| } | } | ||||
| if (type == nullptr) { | |||||
| type_proto->set_data_type(debugger::DT_UNDEFINED); | |||||
| } else if (type->isa<Number>()) { | |||||
| if (type->isa<Number>()) { | |||||
| type_proto->set_data_type(GetDebuggerNumberDataType(type)); | type_proto->set_data_type(GetDebuggerNumberDataType(type)); | ||||
| } | } | ||||
| } | } | ||||
| @@ -52,9 +50,11 @@ void DebuggerProtoExporter::SetNodeOutputType(const TypePtr &type, const BaseSha | |||||
| if (type_proto == nullptr) { | if (type_proto == nullptr) { | ||||
| return; | return; | ||||
| } | } | ||||
| if (type != nullptr) { | |||||
| CheckIfValidType(type, type_proto); | |||||
| if (type == nullptr) { | |||||
| type_proto->set_data_type(debugger::DT_UNDEFINED); | |||||
| return; | |||||
| } | } | ||||
| CheckIfValidType(type, type_proto); | |||||
| if (type->isa<TensorType>()) { | if (type->isa<TensorType>()) { | ||||
| TypePtr elem_type = dyn_cast<TensorType>(type)->element(); | TypePtr elem_type = dyn_cast<TensorType>(type)->element(); | ||||
| type_proto->mutable_tensor_type()->set_elem_type(GetDebuggerNumberDataType(elem_type)); | type_proto->mutable_tensor_type()->set_elem_type(GetDebuggerNumberDataType(elem_type)); | ||||