saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix deleting registry at right time.

Browse source
This commit is contained in:
Ryan Hill 2021-04-24 01:15:14 -07:00
parent 06eac846b8
commit 393286f484
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
onnxruntime/core/providers/cuda/cuda_provider_factory.cc
View file

@ -19,6 +19,8 @@ using namespace onnxruntime;
namespace onnxruntime {
void Shutdown_DeleteRegistry();
struct CUDAProviderFactory : IExecutionProviderFactory {
CUDAProviderFactory(const CUDAExecutionProviderInfo& info)
: info_{info} {}
@ -157,6 +159,7 @@ struct CUDA_Provider : Provider {
}
void Shutdown() override {
Shutdown_DeleteRegistry();
}
} g_provider;