From 3e7dcdc123266687c354350a4b11427869925429 Mon Sep 17 00:00:00 2001 From: lichun Date: Fri, 12 Mar 2021 11:33:24 +0800 Subject: [PATCH] dynamic shape inference support --- tests/ut/ge/generator/ge_generator_unittest.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/ut/ge/generator/ge_generator_unittest.cc b/tests/ut/ge/generator/ge_generator_unittest.cc index 022fd611..5cf5038b 100644 --- a/tests/ut/ge/generator/ge_generator_unittest.cc +++ b/tests/ut/ge/generator/ge_generator_unittest.cc @@ -77,7 +77,7 @@ TEST_F(UtestGeGenerator, test_set_model_name) { GeGenerator generator; generator.Initialize({}); GeRootModelPtr ge_root_model = make_shared(GeRootModel()); - ComputeGraphPtr graph = make_shared(ComputeGraph()); + ComputeGraphPtr graph = make_shared(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);