Browse Source

fix ut

pull/2052/head
guopeian 3 years ago
parent
commit
5682addc4f
1 changed files with 10 additions and 3 deletions
  1. +10
    -3
      tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc

+ 10
- 3
tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc View File

@@ -178,6 +178,13 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_tf_node_task) {
}
TEST_F(UtestAicpuNodeExecutor, aicpu_memcopy_task) {
ComputeGraphPtr graph = std::make_shared<ComputeGraph>("test");
GeModelPtr ge_sub_model = std::make_shared<GeModel>();
GeRootModelPtr ge_root_model = std::make_shared<GeRootModel>(graph);
ge_root_model->SetModelName("test_name");
ge_root_model->SetSubgraphInstanceNameToModel("sub", ge_sub_model);
HybridModel hybrid_model(ge_root_model);
NodePtr node = CreateNode(graph, "frameworkop", FRAMEWORK_OP_TYPE, 4, 2);
std::unique_ptr<NodeItem> new_node;
ASSERT_EQ(NodeItem::Create(node, new_node), SUCCESS);
NodeItem *node_item = new_node.get();
@@ -193,13 +200,13 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_memcopy_task) {
node_item->num_outputs = 0;
AicpuNodeTask aicpu_node_task(node_item, task_def);
ASSERT_EQ(aicpu_nod_task.SetMemCopyTask(task_def), SUCCESS);
aicpu_node_task->node_item_->num_outputs = 1;
aicpu_node_task.node_item_->num_outputs = 1;
kernel_def->set_args_size(0);
ASSERT_EQ(aicpu_nod_task.SetMemCopyTask(task_def), FAILED);
ASSERT_EQ(aicpu_node_task.SetMemCopyTask(task_def), FAILED);
char* args2 = "123";
kernel_def->set_args(reinterpret_cast<const char *>(&args2), 3);
kernel_def->set_args_size(3);
ASSERT_EQ(aicpu_nod_task.SetMemCopyTask(task_def), FAILED);
ASSERT_EQ(aicpu_node_task.SetMemCopyTask(task_def), FAILED);
}
TEST_F(UtestAicpuNodeExecutor, aicpu_blocking_node_task) {


Loading…
Cancel
Save