fix: 添加资源释放幂等保护

- 添加 _released 标志防止重复释放
- release() 方法支持幂等调用
This commit is contained in:
2026-01-30 10:48:28 +08:00
parent b67bda8042
commit 6dc3442cc2

View File

@@ -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引擎资源已释放")