|
|
|
@@ -76,7 +76,7 @@ TEST_F(UtestGeGenerator, test_build_single_op_online) { |
|
|
|
TEST_F(UtestGeGenerator, test_set_model_name) { |
|
|
|
GeGenerator generator; |
|
|
|
generator.Initialize({}); |
|
|
|
GeRootModelPtr ge_root_model = make_shared<GeRootModelPtr>(GeRootModel()); |
|
|
|
GeRootModelPtr ge_root_model = make_shared<GeRootModel>(GeRootModel()); |
|
|
|
ComputeGraphPtr graph = make_shared<ComputeGraph>(ComputeGraph("graph")); |
|
|
|
(void)AttrUtils::SetBool(graph, "_dynamic_shape_partitioned", true); |
|
|
|
ge_root_model->root_graph_ = std::move(graph); |
|
|
|
|