|
|
@@ -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 |