From 75bcf24692072918d51963f02d29502cce70c951 Mon Sep 17 00:00:00 2001 From: zhangxiaokun Date: Fri, 19 Feb 2021 16:31:29 +0800 Subject: [PATCH] Add unit test for no NetOutput --- tests/ut/ge/graph/load/davinci_model_unittest.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/ut/ge/graph/load/davinci_model_unittest.cc b/tests/ut/ge/graph/load/davinci_model_unittest.cc index f10068aa..b8a963e3 100644 --- a/tests/ut/ge/graph/load/davinci_model_unittest.cc +++ b/tests/ut/ge/graph/load/davinci_model_unittest.cc @@ -345,7 +345,7 @@ TEST_F(UtestDavinciModel, Init_variable_op) { EXPECT_EQ(model.InitNodes(graph), SUCCESS); - EXPECT_EQ(model.ReturnNoOutput(1), PARAM_INVALID); + EXPECT_EQ(model.ReturnNoOutput(1), SUCCESS); EXPECT_EQ(model.SyncVarData(), SUCCESS); OutputData output_data;