saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-06 00:03:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Removing one of the generate tranform tests (#1010)

Browse source
This commit is contained in:
Konstantinos Karanasos 2019-05-12 18:01:08 -07:00 committed by jignparm
parent c0acb8b6c3
commit 4e5fdd3167
1 changed files with 0 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -34,25 +34,6 @@ TEST(GraphTransformerUtilsTests, TestGenerateRewriterules) {
}
TEST(GraphTransformerUtilsTests, TestGenerateGraphTransformers) {
auto transformers = transformer_utils::GenerateTransformers(TransformerLevel::Level2);
ASSERT_TRUE(transformers.size() != 0);
// Transformer name match test
std::vector<std::string> custom_list = {"EliminateIdentity", "GemmActivationFusion", "MatMulAddFusion", "abc", "def"};
transformers = transformer_utils::GenerateTransformers(TransformerLevel::Level2, custom_list);
ASSERT_TRUE(transformers.size() == 2);
// validate each rule returned is present in the custom list
for (const auto& transformer : transformers) {
ASSERT_TRUE(std::find(custom_list.begin(), custom_list.end(), transformer->Name()) != custom_list.end());
}
// Transformer name no-match test. When there is no match, empty list is expected.
custom_list = {"EliminateIdentity"};
transformers = transformer_utils::GenerateTransformers(TransformerLevel::Level2, custom_list);
ASSERT_TRUE(transformers.size() == 0);
}
TEST(GraphTransformerUtilsTests, TestGenerateGraphTransformers_CustomList) {
// custom list of rules and transformers
std::string l1_rule1 = "EliminateIdentity";
std::string l1_transformer = "ConstantFolding";