Browse Source

modify error_code related by op

pull/2072/head
yuxing 3 years ago
parent
commit
312abfca62
1 changed files with 6 additions and 3 deletions
  1. +6
    -3
      tests/ut/ge/generator/ge_generator_unittest.cc

+ 6
- 3
tests/ut/ge/generator/ge_generator_unittest.cc View File

@@ -188,9 +188,12 @@ TEST_F(UtestGeGenerator, test_generate_online_model) {
}

TEST_F(UtestGeGenerator, test_check_engine_type_support) {
GeTensorDesc tensor_desc;
shared_ptr<OpDesc> op_desc = make_shared<OpDesc>("Add", "add");
GeGenerator generator;
generator.Initialize({});
EXPECT_EQ(generator.CheckEngineTypeSupport(op_desc, ENGINE_AIVECTOR), FAILED);
op_desc->AddInputDesc(tensor_desc);
op_desc->AddInputDesc(tensor_desc);
op_desc->AddOutputDesc(tensor_desc);
EXPECT_EQ(CheckEngineTypeSupport(op_desc, ENGINE_AIVECTOR), FAILED);
}
} // namespace ge

Loading…
Cancel
Save