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

Fast exit workaround for dnnl test. (#3863)

Browse source
This commit is contained in:
Ryan Hill 2020-05-07 11:23:40 -07:00 committed by GitHub
parent 0aeb383273
commit 3bafc096bb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
onnxruntime/test/onnx/main.cc
View file

@ -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;
}