From 068504e8eddc52445538e0e779a015d2e88f73ec Mon Sep 17 00:00:00 2001
From: simson <wudengsong@huawei.com>
Date: Mon, 2 Nov 2020 17:03:57 +0800
Subject: [PATCH] add uint for SetValueToProto

---
 mindspore/ccsrc/debug/dump_proto.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/mindspore/ccsrc/debug/dump_proto.cc b/mindspore/ccsrc/debug/dump_proto.cc
index a6c4ca8767..84e166311c 100644
--- 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);