fix: 添加资源释放幂等保护
- 添加 _released 标志防止重复释放 - release() 方法支持幂等调用
This commit is contained in:
@@ -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引擎资源已释放")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user