Browse Source

fix ut

pull/2039/head
guopeian 3 years ago
parent
commit
752da1b1da
2 changed files with 1 additions and 2 deletions
  1. +0
    -1
      tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc
  2. +1
    -1
      tests/ut/ge/single_op/single_op_model_unittest.cc

+ 0
- 1
tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc View File

@@ -154,7 +154,6 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_tf_node_task) {
AicpuExtInfo aicpu_ext_info2;
aicpu_ext_info2.infoType = aicpu::FWKAdapter::FWK_ADPT_EXT_SHAPE_TYPE;
aicpu_ext_info2.infoLen = sizeof(int32_t);
int32_t type = node_item->shape_inference_type;
memcpy_s(aicpu_ext_info2.infoMsg, sizeof(int32_t), &type, sizeof(int32_t));
char *ext_mem2 = (char*)malloc(sizeof(AicpuExtInfo) + sizeof(int32_t));
memcpy_s(ext_mem2, sizeof(AicpuExtInfo) + sizeof(int32_t), &aicpu_ext_info2, sizeof(AicpuExtInfo) + sizeof(int32_t));


+ 1
- 1
tests/ut/ge/single_op/single_op_model_unittest.cc View File

@@ -394,7 +394,7 @@ TEST_F(UtestSingleOpModel, build_dynamic_task02) {
StreamResource *res = new (std::nothrow) StreamResource(1);

ASSERT_EQ(model.ParseTasks(), SUCCESS);
model.node_tasks_[node] = { *task_def2 };
model.node_tasks_[node] = { *task_def, *task_def };
op_desc->SetOpKernelLibName(kEngineNameAiCpu);
model.BuildTaskListForDynamicOp(res, single_op);
}

Loading…
Cancel
Save