From 3bafc096bb42ddbb9dc252d052ba35f44d642804 Mon Sep 17 00:00:00 2001
From: Ryan Hill <38674843+RyanUnderhill@users.noreply.github.com>
Date: Thu, 7 May 2020 11:23:40 -0700
Subject: [PATCH] Fast exit workaround for dnnl test. (#3863)

---
 onnxruntime/test/onnx/main.cc | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/onnxruntime/test/onnx/main.cc b/onnxruntime/test/onnx/main.cc
index 3baa1dc51b..29adeb8070 100644
--- a/onnxruntime/test/onnx/main.cc
+++ b/onnxruntime/test/onnx/main.cc
@@ -24,6 +24,8 @@
 
 using namespace onnxruntime;
 
+static bool g_exit_fast = false;
+
 namespace {
 void usage() {
   printf(
@@ -321,6 +323,7 @@ int real_main(int argc, char* argv[], Ort::Env& env) {
     if (enable_dnnl) {
 #ifdef USE_DNNL
       Ort::ThrowOnError(OrtSessionOptionsAppendExecutionProvider_Dnnl(sf, enable_cpu_mem_arena ? 1 : 0));
+      g_exit_fast = true;
 #else
       fprintf(stderr, "DNNL is not supported in this build");
       return -1;
@@ -855,6 +858,7 @@ int main(int argc, char* argv[]) {
     retval = -1;
   }
   ::google::protobuf::ShutdownProtobufLibrary();
-  std::cout << "*** Exiting Test Runner\r\n";
+  if (g_exit_fast)
+    std::_Exit(retval);
   return retval;
 }