From ef3e760e75770b05e420d4038cb32b644e83a0ea Mon Sep 17 00:00:00 2001 From: unknown Date: Sun, 28 Feb 2021 17:18:22 +0800 Subject: [PATCH] Add single_op model_id. --- tests/ut/ge/single_op/stream_resource_unittest.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/ut/ge/single_op/stream_resource_unittest.cc b/tests/ut/ge/single_op/stream_resource_unittest.cc index 8a5124ef..2138e2c9 100644 --- a/tests/ut/ge/single_op/stream_resource_unittest.cc +++ b/tests/ut/ge/single_op/stream_resource_unittest.cc @@ -63,11 +63,11 @@ TEST_F(UtestStreamResource, test_malloc_memory) { ModelData model_data; SingleOp *single_op = nullptr; DynamicSingleOp *dynamic_single_op = nullptr; - res.op_map_[0] = &single_op; - res.dynamic_op_map_[1] = &dynamic_single_op; + res.op_map_[0].reset(single_op); + res.dynamic_op_map_[1].reset(dynamic_single_op); - ASSERT_EQ(res.BuildOperator(model_data, &single_op, 0), SUCCESS); - ASSERT_EQ(res.BuildDynamicOperator(model_data, &dynamic_single_op, 1), SUCCESS); + EXPECT_EQ(res.BuildOperator(model_data, &single_op, 0), SUCCESS); + EXPECT_EQ(res.BuildDynamicOperator(model_data, &dynamic_single_op, 1), SUCCESS); } /*