|
@@ -152,6 +152,20 @@ TEST_F(UtestGeHybrid, index_taskdefs_failed) { |
|
|
ASSERT_EQ(hybrid_model_builder.IndexTaskDefs(graph, ge_model), INTERNAL_ERROR); |
|
|
ASSERT_EQ(hybrid_model_builder.IndexTaskDefs(graph, ge_model), INTERNAL_ERROR); |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
TEST_F(UtestGeHybrid, parse_force_infershape_nodes) { |
|
|
|
|
|
const char *const kForceInfershape = "_force_infershape_when_running"; |
|
|
|
|
|
auto graph = make_shared<ComputeGraph>("graph"); |
|
|
|
|
|
OpDescPtr op_desc = CreateOpDesc("Conv2D", "Conv2D"); |
|
|
|
|
|
ge::AttrUtils::SetBool(op_desc, kForceInfershape, true); |
|
|
|
|
|
auto node = graph->AddNode(op_desc); |
|
|
|
|
|
std::unique_ptr<NodeItem> new_node; |
|
|
|
|
|
NodeItem::Create(node, new_node); |
|
|
|
|
|
GeRootModelPtr ge_root_model = make_shared<GeRootModel>(graph); |
|
|
|
|
|
HybridModel hybrid_model(ge_root_model); |
|
|
|
|
|
HybridModelBuilder hybrid_model_builder(hybrid_model); |
|
|
|
|
|
ASSERT_EQ(hybrid_model_builder.ParseForceInfershapeNodes(node, *new_node), SUCCESS); |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
TEST_F(UtestGeHybrid, index_taskdefs_success) { |
|
|
TEST_F(UtestGeHybrid, index_taskdefs_success) { |
|
|
// build aicore task |
|
|
// build aicore task |
|
|
domi::ModelTaskDef model_task_def; |
|
|
domi::ModelTaskDef model_task_def; |
|
|