Browse Source

Add single_op model_id.

tags/v1.2.0
unknown 4 years ago
parent
commit
ef3e760e75
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      tests/ut/ge/single_op/stream_resource_unittest.cc

+ 4
- 4
tests/ut/ge/single_op/stream_resource_unittest.cc View File

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

/*


Loading…
Cancel
Save