|
|
@@ -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) {
|
|
|
|