saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Avoid using transformers that rely on contrib ops in tests (#983)

Browse source
This commit is contained in:
Konstantinos Karanasos 2019-05-07 17:23:04 -07:00 committed by GitHub
parent 17690355ed
commit 403fb703a0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
onnxruntime/test/optimizer/graph_transform_utils_test.cc
View file

@ -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<std::string> custom_list = {l1_rule1, l1_transformer, l2_transformer};
std::string l2_rule1 = "ConvBNFusion";
std::vector<std::string> 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<RuleBasedGraphTransformer*>(transformers[0].get());
ASSERT_TRUE(rule_transformer->RulesCount() == 1);
}
} // namespace test