Browse Source

Add ut.

tags/v1.3.0
zhaozhixuan 3 years ago
parent
commit
6af2a247fd
2 changed files with 10 additions and 1 deletions
  1. +1
    -1
      ge/generator/ge_generator.cc
  2. +9
    -0
      tests/ut/ge/generator/ge_generator_unittest.cc

+ 1
- 1
ge/generator/ge_generator.cc View File

@@ -714,7 +714,7 @@ void GeGenerator::RemoveConst(const vector<GeTensor> &inputs, vector<GeTensor> &
for (auto input : inputs) { for (auto input : inputs) {
GeTensorDesc input_desc = input.GetTensorDesc(); GeTensorDesc input_desc = input.GetTensorDesc();
bool is_const = false; bool is_const = false;
(void)AttrUtils::GetBool(tensor, CONST_ATTR_NAME_INPUT, is_const);
(void)AttrUtils::GetBool(input_desc, CONST_ATTR_NAME_INPUT, is_const);
if (!is_const) { if (!is_const) {
outputs.emplace_back(input); outputs.emplace_back(input);
} }


+ 9
- 0
tests/ut/ge/generator/ge_generator_unittest.cc View File

@@ -128,4 +128,13 @@ TEST_F(UtestGeGenerator, test_set_model_name) {
ge_root_model->root_graph_ = std::move(graph); ge_root_model->root_graph_ = std::move(graph);
EXPECT_EQ(generator.SetModelNameForDump(ge_root_model), SUCCESS); EXPECT_EQ(generator.SetModelNameForDump(ge_root_model), SUCCESS);
} }

TEST_F(UtestGeGenerator, test_remove_const) {
GeGenerator generator;
GeTensorDesc tensor_desc;
GeTensor tensor(tensor_desc);
const vector<GeTensor> inputs = {tensor};
vector<GeTensor> outputs;
generator.RemoveConst(inputs, outputs);
}
} // namespace ge } // namespace ge

Loading…
Cancel
Save