From 07e9f500da0d01b984f3d4e41770957bca06b080 Mon Sep 17 00:00:00 2001 From: Hariharan Seshadri Date: Sun, 20 Oct 2019 17:13:02 -0700 Subject: [PATCH] Add back Keras Mask RCNN to CUDA test exclusion (#2201) --- onnxruntime/test/onnx/main.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/onnxruntime/test/onnx/main.cc b/onnxruntime/test/onnx/main.cc index f3115be11c..8d82ff725c 100644 --- a/onnxruntime/test/onnx/main.cc +++ b/onnxruntime/test/onnx/main.cc @@ -521,6 +521,7 @@ int real_main(int argc, char* argv[], Ort::Env& env) { #endif #ifdef USE_CUDA + broken_tests.insert({"mask_rcnn_keras", "result mismatch"}); broken_tests.insert({"mlperf_ssd_mobilenet_300", "unknown error"}); broken_tests.insert({"mlperf_ssd_resnet34_1200", "unknown error"}); broken_tests.insert({"tf_inception_v1", "flaky test"}); //TODO: Investigate cause for flakiness