We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 04188a8 commit 6f83b41Copy full SHA for 6f83b41
core/conversion/conversionctx/ConversionCtx.cpp
@@ -148,7 +148,9 @@ std::string ConversionCtx::SerializeEngine() {
148
auto engine = builder->buildEngineWithConfig(*net, *cfg);
149
auto serialized_engine = engine->serialize();
150
engine->destroy();
151
- return std::string((const char*)serialized_engine->data(), serialized_engine->size());
+ auto engine_str = std::string((const char*)serialized_engine->data(), serialized_engine->size());
152
+ serialized_engine->destroy();
153
+ return engine_str;
154
}
155
156
bool ConversionCtx::CheckLayerAddition(const torch::jit::Node* n) {
0 commit comments