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

Exclude tests that fail when contrib ops are disabled. (#835)

Browse source
This commit is contained in:
Pranav Sharma 2019-04-15 15:57:48 -07:00 committed by GitHub
parent 4b4a359943
commit 049ba2d747
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/test/optimizer/graph_transform_test.cc
View file

@ -148,6 +148,7 @@ TEST(GraphTransformationTests, FuseConvBNMulAddUnsqueeze) {
}
}
#ifndef DISABLE_CONTRIB_OPS
TEST(GraphTransformationTests, FuseConvActivation) {
std::unordered_map<std::string, std::string> model_to_op_name{{"fusion/conv_relu.onnx", "Relu"},
{"fusion/conv_sigmoid.onnx", "Sigmoid"},
@ -172,6 +173,7 @@ TEST(GraphTransformationTests, FuseConvActivation) {
ASSERT_TRUE(op_to_count[model.second] == 0);
}
}
#endif
TEST(GraphTransformationTests, FuseConvMulNoBias) {
string model_uri = MODEL_FOLDER + "fusion/fuse-conv-mul-no-bias.onnx";
@ -269,6 +271,7 @@ TEST(GraphTransformationTests, MatMulAddFusion_three_input) {
ASSERT_TRUE(op_to_count["Gemm"] == 1);
}
#ifndef DISABLE_CONTRIB_OPS
TEST(GraphTransformationTests, Gemm_Relu_three_input) {
string model_uri = MODEL_FOLDER + "matmul_add_fusion/3Input/gemm_relu.onnx";
@ -283,6 +286,7 @@ TEST(GraphTransformationTests, Gemm_Relu_three_input) {
std::map<std::string, int> op_to_count = CountOpsInGraph(graph);
ASSERT_TRUE(op_to_count["Relu"] == 0);
}
#endif
TEST(GraphTransformationTests, FuseConvBnAddMulFloat16) {
string model_uri = MODEL_FOLDER + "fusion/fuse-conv-bn-add-mul-float16.onnx";