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