From 30ed06b9ec45b9cef223fe6a0582d0c1789f9464 Mon Sep 17 00:00:00 2001 From: Baiju Meswani Date: Fri, 27 Sep 2024 17:24:19 -0700 Subject: [PATCH] Keep arena config alive --- src/generators.cpp | 6 +++--- src/generators.h | 1 + 2 files changed, 4 insertions(+), 3 deletions(-) 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_;