!8129 add uint for SetValueToProto

Merge pull request !8129 from Simson/bugfix
4 years ago · 06838ded82
parent e9cd12e904 068504e8ed
commit 06838ded82
1 changed files with 3 additions and 0 deletions
--- a/mindspore/ccsrc/debug/dump_proto.cc
+++ b/mindspore/ccsrc/debug/dump_proto.cc
@ -185,6 +185,9 @@ void ProtoExporter::SetValueToProto(const ValuePtr &val, irpb::ValueProto *value
  } else if (val->isa<Int>()) {
    value_proto->set_dtype(irpb::DT_TYPE);
    value_proto->mutable_type_val()->set_data_type(irpb::DT_BASE_INT);
+  } else if (val->isa<UInt>()) {
+    value_proto->set_dtype(irpb::DT_TYPE);
+    value_proto->mutable_type_val()->set_data_type(irpb::DT_BASE_UINT);
  } else if (val->isa<Float>()) {
    value_proto->set_dtype(irpb::DT_TYPE);
    value_proto->mutable_type_val()->set_data_type(irpb::DT_BASE_FLOAT);