saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-26 22:35:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Diagnostics

Browse source
This commit is contained in:
Ryan Hill 2021-04-23 13:39:49 -07:00
parent 5c6910ed9c
commit 96fa0845a2
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
onnxruntime/core/framework/provider_bridge_ort.cc
View file

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