From de50397b5aa4d4b472b2574a36088dbc066f6d63 Mon Sep 17 00:00:00 2001 From: guopeian Date: Mon, 19 Jul 2021 19:06:47 +0800 Subject: [PATCH] fix --- tests/ut/ge/single_op/single_op_model_unittest.cc | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/ut/ge/single_op/single_op_model_unittest.cc b/tests/ut/ge/single_op/single_op_model_unittest.cc index 75db349e..ee2d2c10 100644 --- a/tests/ut/ge/single_op/single_op_model_unittest.cc +++ b/tests/ut/ge/single_op/single_op_model_unittest.cc @@ -410,6 +410,13 @@ TEST_F(UtestSingleOpModel, build_dynamic_task02) { model.model_helper_.model_ = ge_model; auto op_desc = std::make_shared("add", "Add"); AttrUtils::SetInt(op_desc, ::ge::ATTR_NAME_UNKNOWN_SHAPE_TYPE, ge::DEPEND_COMPUTE); + GeTensorDesc tensor(GeShape(), FORMAT_NCHW, DT_FLOAT); + for (int i = 0; i < 4; i++) { + op_desc->AddInputDesc(tensor); + } + for (int i = 0; i < 2; i++) { + op_desc->AddOutputDesc(tensor); + } NodePtr node = graph->AddNode(op_desc); model.op_list_[0] = node; StreamResource *res = new (std::nothrow) StreamResource(1);