Fix error handling for OnSessionInitializationEnd

This commit is contained in:
Jeff 2020-04-15 15:37:13 -07:00
parent e89dd92387
commit d18f0fc47b

View file

@ -931,12 +931,14 @@ common::Status InferenceSession::Initialize() {
}
if (status.IsOK()) {
auto retval = status;
for (auto& xp : execution_providers_) {
auto status = xp->OnSessionInitializationEnd();
ORT_CHECK_AND_SET_RETVAL(status);
auto endStatus = xp->OnSessionInitializationEnd();
if (status.IsOK()) {
status = endStatus;
}
}
}
return status;
}