From 716a79d8d7984173e419179e1ccad4c949b55dea Mon Sep 17 00:00:00 2001 From: zhaozhixuan Date: Fri, 30 Apr 2021 07:31:28 +0800 Subject: [PATCH] run_graph_with_stream. --- tests/ut/ge/graph/ge_executor_unittest.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/ut/ge/graph/ge_executor_unittest.cc b/tests/ut/ge/graph/ge_executor_unittest.cc index 610fe98b..3c6a9903 100644 --- a/tests/ut/ge/graph/ge_executor_unittest.cc +++ b/tests/ut/ge/graph/ge_executor_unittest.cc @@ -318,9 +318,9 @@ TEST_F(UtestGeExecutor, get_op_attr) { auto ret = ge_executor.GetOpAttr(1, "test", ge::ATTR_NAME_DATA_DUMP_ORIGIN_OP_NAMES, attr_value); EXPECT_EQ(ret, SUCCESS); EXPECT_EQ(attr_value, "[4]test"); - // ret = ge_executor.GetOpAttr(2, "test", ge::ATTR_NAME_DATA_DUMP_ORIGIN_OP_NAMES, attr_value); - // EXPECT_EQ(ret, UNSUPPORTED); - // ret = ge_executor.GetOpAttr(3, "test", ge::ATTR_NAME_DATA_DUMP_ORIGIN_OP_NAMES, attr_value); - // EXPECT_EQ(ret, ACL_ERROR_GE_EXEC_MODEL_ID_INVALID); + ret = ge_executor.GetOpAttr(2, "test", ge::ATTR_NAME_DATA_DUMP_ORIGIN_OP_NAMES, attr_value); + EXPECT_EQ(ret, PARAM_INVALID); + ret = ge_executor.GetOpAttr(3, "test", ge::ATTR_NAME_DATA_DUMP_ORIGIN_OP_NAMES, attr_value); + EXPECT_EQ(ret, ACL_ERROR_GE_EXEC_MODEL_ID_INVALID); } } \ No newline at end of file