From 6dc3442cc247e7e8530790f5a562d63455179d23 Mon Sep 17 00:00:00 2001 From: 16337 <1633794139@qq.com> Date: Fri, 30 Jan 2026 10:48:28 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E6=B7=BB=E5=8A=A0=E8=B5=84=E6=BA=90?= =?UTF-8?q?=E9=87=8A=E6=94=BE=E5=B9=82=E7=AD=89=E4=BF=9D=E6=8A=A4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 添加 _released 标志防止重复释放 - release() 方法支持幂等调用 --- core/tensorrt_engine.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/core/tensorrt_engine.py b/core/tensorrt_engine.py index 15a3663..37e832f 100644 --- a/core/tensorrt_engine.py +++ b/core/tensorrt_engine.py @@ -49,6 +49,7 @@ class TensorRTEngine: self._input_binding = None self._output_bindings = [] self._stream = None + self._released = False self._logger = get_logger("tensorrt") self._lock = threading.Lock() @@ -348,8 +349,12 @@ class TensorRTEngine: self._memory_pool.clear() def release(self): - """释放引擎资源""" + """释放引擎资源(幂等调用)""" with self._lock: + if self._released: + return + self._released = True + self._release_resources() self._logger.info("TensorRT引擎资源已释放")