experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[TF:TRT] Fix memory leakage in TRTEngineOp::ExecuteNativeSegment.

Browse Source 
The statement `delete outputs` may not be executed in the original implementation.

PiperOrigin-RevId: 347915990
Change-Id: Ic5cee6b19eb55ce5da5b3669ede0059d1889288c
This commit is contained in:
A. Unique TensorFlower 2020-12-16 16:04:54 -08:00 committed by TensorFlower Gardener
parent 475edf87f4
commit d7f7f61c76
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/tf2tensorrt/kernels/trt_engine_op.cc
View File

@ -461,12 +461,12 @@ void TRTEngineOp::ExecuteNativeSegment(OpKernelContext* ctx,
lib->Run(opts, native_execution_func_handle_, inputs, outputs, lib->Run(opts, native_execution_func_handle_, inputs, outputs,
[this, ctx, outputs, helper](const Status& s) { [this, ctx, outputs, helper](const Status& s) {
core::ScopedUnref sc(helper); core::ScopedUnref sc(helper);
std::unique_ptr<std::vector<Tensor>> outputs_wrapper(outputs);
OP_REQUIRES_OK_ASYNC(ctx, s, *helper); OP_REQUIRES_OK_ASYNC(ctx, s, *helper);
VLOG(1) << "Native Segment completed"; VLOG(1) << "Native Segment completed";
for (size_t t = 0; t < outputs->size(); ++t) { for (size_t t = 0; t < outputs->size(); ++t) {
ctx->set_output(t, outputs->at(t)); ctx->set_output(t, outputs->at(t));
} }
delete outputs;
}); });
} }