Fix "./onnxruntime_test_all --help" segfault #22839
Conversation
| ::testing::InitGoogleTest(&argc, argv); | ||
| ortenv_setup(); |
There was a problem hiding this comment.
This is very tricky. The ::testing::InitGoogleTest function also calls a lot of initialization functions in the tests, and the these functions may call into ONNX Runtime. Therefore ortenv should be initialized before that.
There was a problem hiding this comment.
I see, I pushed an alternative restoring the original order, but making sure ortenv teardown is always done. This one also seems working with "--help" case
| @@ -93,6 +93,8 @@ int TEST_MAIN(int argc, char** argv) { | |||
|
|
|||
| ORT_TRY { | |||
| ortenv_setup(); | |||
| // TODO: Fix the C API issue | |||
| atexit(ortenv_teardown); // If we don't do this, it will crash | |||
There was a problem hiding this comment.
Atexit functions get executed after the main function is returned, especially , it is after the "::google::protobuf::ShutdownProtobufLibrary" , which means in the teardown function we cannot call all protobuf functions, which is not possible. Also, it will also change the destruction order between function local statics and this "ortenv_teardown" function. Because basically the function local statics will use atexit function to register their destructors. Therefore, here the ordering is really very tricky. I would suggest keeping it unchanged unless there is an urgent need.
There was a problem hiding this comment.
would using gsl::finally instead of atexit work?
e.g., #include <gsl/gsl>, then,
| atexit(ortenv_teardown); // If we don't do this, it will crash | |
| auto clean_up_ortenv = gsl::finally(ortenv_teardown); // If we don't do this, it will crash |
Description
Makes sure the GTest parameters are processed before ORT Env is set up in unittests binary. In this way, binary won't segfault on exit if --help arg is used
Motivation and Context
Fixes #22838