From 16c5281fbb64ed52a2d097981f7b71df4c8591a9 Mon Sep 17 00:00:00 2001
From: chuxing <chuxing@huawei.com>
Date: Sun, 28 Mar 2021 18:44:21 +0800
Subject: [PATCH] update UT

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

diff --git a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
index 4f10e6e4..6a0e974e 100644
--- a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
+++ b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
@@ -361,6 +361,9 @@ TEST_F(UtestGeHybrid, TestTaskContext) {
   desc = task_context->MutableOutputDesc(0);
   ASSERT_TRUE(desc != nullptr);
   ASSERT_EQ(desc->GetShape().GetDims(), shape.GetDims());
+  GeTensorDesc output_desc;
+  ASSERT_EQ(task_context->GetOutputDesc(0, output_desc), SUCCESS);
+  ASSERT_EQ(output_desc.GetShape().GetDims(), shape.GetDims());
 
   desc = task_context->MutableInputDesc(0);
   ASSERT_TRUE(desc != nullptr);