From 1307f203c59d84fdd3a7f4de255887f3ba3ee2a6 Mon Sep 17 00:00:00 2001
From: xutianchun <xutianchun@huawei.com>
Date: Wed, 24 Mar 2021 17:27:40 +0800
Subject: [PATCH] fix weight quant may print error

---
 mindspore/lite/tools/converter/quantizer/weight_quantizer.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/lite/tools/converter/quantizer/weight_quantizer.cc b/mindspore/lite/tools/converter/quantizer/weight_quantizer.cc
index 68706d6df5..1b1902c29d 100644
--- a/mindspore/lite/tools/converter/quantizer/weight_quantizer.cc
+++ b/mindspore/lite/tools/converter/quantizer/weight_quantizer.cc
@@ -672,7 +672,7 @@ STATUS WeightQuantizer::DoMixedQuant(FuncGraphPtr func_graph) {
 STATUS WeightQuantizer::DoFixedQuant(FuncGraphPtr func_graph) {
   MS_ASSERT(func_graph != nullptr);
   for (auto &cnode : func_graph->GetOrderedCnodes()) {
-    auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0));
+    auto primitive = GetValueNode<std::shared_ptr<ops::PrimitiveC>>(cnode->input(0));
     if (primitive == nullptr) {
       MS_LOG(DEBUG) << cnode->fullname_with_scope() << " : primitive is nullptr";
       continue;