Skip to content

Commit 6f83b41

Browse files
committed
fix(//core/conversion/conversionctx): Fix memory leak in conversion
context engine serialization h/t: @SakodaShintaro Signed-off-by: Naren Dasan <naren@narendasan.com> Signed-off-by: Naren Dasan <narens@nvidia.com>
1 parent 04188a8 commit 6f83b41

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

core/conversion/conversionctx/ConversionCtx.cpp

+3-1
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,9 @@ std::string ConversionCtx::SerializeEngine() {
148148
auto engine = builder->buildEngineWithConfig(*net, *cfg);
149149
auto serialized_engine = engine->serialize();
150150
engine->destroy();
151-
return std::string((const char*)serialized_engine->data(), serialized_engine->size());
151+
auto engine_str = std::string((const char*)serialized_engine->data(), serialized_engine->size());
152+
serialized_engine->destroy();
153+
return engine_str;
152154
}
153155

154156
bool ConversionCtx::CheckLayerAddition(const torch::jit::Node* n) {

0 commit comments

Comments
 (0)