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);