From 96fa0845a2fa2fc2479a046bd78e3aa9b041209b Mon Sep 17 00:00:00 2001 From: Ryan Hill Date: Fri, 23 Apr 2021 13:39:49 -0700 Subject: [PATCH] Diagnostics --- onnxruntime/core/framework/provider_bridge_ort.cc | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/onnxruntime/core/framework/provider_bridge_ort.cc b/onnxruntime/core/framework/provider_bridge_ort.cc index 4810652dce..e3e4bdfb89 100644 --- a/onnxruntime/core/framework/provider_bridge_ort.cc +++ b/onnxruntime/core/framework/provider_bridge_ort.cc @@ -945,11 +945,17 @@ static ProviderLibrary s_library_openvino(LIBRARY_PREFIX "onnxruntime_providers_ static ProviderLibrary s_library_tensorrt(LIBRARY_PREFIX "onnxruntime_providers_tensorrt" LIBRARY_EXTENSION); void UnloadSharedProviders() { - s_library_cuda.Unload(); + std::cout << "Unloading shared providers..."; + LOGS_DEFAULT(ERROR) << "LOGS_DEFAULT: Unloading shared providers..."; + s_library_dnnl.Unload(); s_library_openvino.Unload(); s_library_tensorrt.Unload(); + s_library_cuda.Unload(); s_library_shared.Unload(); + + std::cout << "Finished Unloading shared providers"; + LOGS_DEFAULT(ERROR) << "LOGS_DEFAULT: Finished Unloading shared providers"; } // Used by test code