From 403fb703a0f9edfc53fe17a7f638f8b883ac3904 Mon Sep 17 00:00:00 2001 From: Konstantinos Karanasos Date: Tue, 7 May 2019 17:23:04 -0700 Subject: [PATCH] Avoid using transformers that rely on contrib ops in tests (#983) --- onnxruntime/test/optimizer/graph_transform_utils_test.cc | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/onnxruntime/test/optimizer/graph_transform_utils_test.cc b/onnxruntime/test/optimizer/graph_transform_utils_test.cc index 63f7e27da1..ebbf91af2d 100644 --- a/onnxruntime/test/optimizer/graph_transform_utils_test.cc +++ b/onnxruntime/test/optimizer/graph_transform_utils_test.cc @@ -56,8 +56,8 @@ TEST(GraphTransformerUtilsTests, TestGenerateGraphTransformers_CustomList) { // custom list of rules and transformers std::string l1_rule1 = "EliminateIdentity"; std::string l1_transformer = "ConstantFolding"; - std::string l2_transformer = "GemmActivationFusion"; - std::vector custom_list = {l1_rule1, l1_transformer, l2_transformer}; + std::string l2_rule1 = "ConvBNFusion"; + std::vector custom_list = {l1_rule1, l1_transformer, l2_rule1}; auto transformers = transformer_utils::GenerateTransformers(TransformerLevel::Level1, custom_list); ASSERT_TRUE(transformers.size() == 2); @@ -72,7 +72,8 @@ TEST(GraphTransformerUtilsTests, TestGenerateGraphTransformers_CustomList) { transformers = transformer_utils::GenerateTransformers(TransformerLevel::Level2, custom_list); ASSERT_TRUE(transformers.size() == 1); - ASSERT_TRUE(transformers[0]->Name() == l2_transformer); + rule_transformer = dynamic_cast(transformers[0].get()); + ASSERT_TRUE(rule_transformer->RulesCount() == 1); } } // namespace test