fix the segmentation fault when open debugger.

pull/12709/head
maning202007 4 years ago
parent 2197f01d0c
commit 815a04cf75

@ -40,9 +40,7 @@ void CheckIfValidType(const TypePtr &type, debugger::TypeProto *type_proto) {
type->isa<UMonadType>() || type->isa<IOMonadType>())) {
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));
}
}
@ -52,9 +50,11 @@ void DebuggerProtoExporter::SetNodeOutputType(const TypePtr &type, const BaseSha
if (type_proto == nullptr) {
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>()) {
TypePtr elem_type = dyn_cast<TensorType>(type)->element();
type_proto->mutable_tensor_type()->set_elem_type(GetDebuggerNumberDataType(elem_type));

Loading…
Cancel
Save