diff --git a/src/generators.cpp b/src/generators.cpp index 36f12a2bd..cc96d2334 100644 --- a/src/generators.cpp +++ b/src/generators.cpp @@ -14,10 +14,10 @@ namespace Generators { static bool _ = (Ort::InitApi(), false); OrtGlobals::OrtGlobals() - : env_{OrtEnv::Create(OrtLoggingLevel::ORT_LOGGING_LEVEL_ERROR)} { - auto arena_config = OrtArenaCfg::Create(0, -1, -1, -1); + : env_{OrtEnv::Create(OrtLoggingLevel::ORT_LOGGING_LEVEL_ERROR)}, + arena_config_{OrtArenaCfg::Create(0, -1, -1, -1)} { Ort::Allocator& allocator_cpu{Ort::Allocator::GetWithDefaultOptions()}; - env_->CreateAndRegisterAllocator(allocator_cpu.GetInfo(), *arena_config); + env_->CreateAndRegisterAllocator(allocator_cpu.GetInfo(), *arena_config_); } // Ensure Shutdown() has been called before process exit diff --git a/src/generators.h b/src/generators.h index 488dd8fa9..757bdfab8 100644 --- a/src/generators.h +++ b/src/generators.h @@ -121,6 +121,7 @@ struct OrtGlobals { OrtGlobals(); std::unique_ptr env_; + std::unique_ptr arena_config_; #if USE_CUDA std::unique_ptr memory_info_cuda_; std::unique_ptr allocator_cuda_;