From 612463e08970dc15eddaf18247a62a17746313c2 Mon Sep 17 00:00:00 2001
From: zhaoxinxin <zhaoxinxin1@huawei.com>
Date: Tue, 9 Mar 2021 20:18:48 +0800
Subject: [PATCH] 	modified:   tests/ut/ge/hybrid/ge_hybrid_unittest.cc

---
 tests/ut/ge/hybrid/ge_hybrid_unittest.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
index 6789f0b1..659d11c6 100644
--- a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
+++ b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
@@ -34,6 +34,7 @@
 #include "hybrid/node_executor/aicore/aicore_task_builder.h"
 #include "graph/load/model_manager/tbe_handle_store.h"
 #include "graph/types.h"
+#include "graph/utils/tensor_utils.h"
 
 #undef private
 #undef protected
@@ -212,7 +213,7 @@ TEST_F(UtestGeHybrid, init_weight_success) {
   ConstGeTensorPtr constTensor_0 =
     std::make_shared<GeTensor>(tensor_desc_0, (uint8_t *)&data_vec_0[0], data_vec_0.size() * sizeof(int32_t));
   AttrUtils::SetTensor(const_op_desc, ge::ATTR_NAME_WEIGHTS, constTensor_0);
-  const_op_desc->AddOutputDesc(constTensor_0);
+  const_op_desc->AddOutputDesc(tensor_desc_0);
   NodePtr const_node = sub_graph->AddNode(const_op_desc);
   graph->AddSubgraph("sub", sub_graph);