Browse Source

update UT

pull/1383/head
chuxing 4 years ago
parent
commit
16c5281fbb
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      tests/ut/ge/hybrid/ge_hybrid_unittest.cc

+ 3
- 0
tests/ut/ge/hybrid/ge_hybrid_unittest.cc View File

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


Loading…
Cancel
Save