We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e024a86 commit d27365dCopy full SHA for d27365d
engine/src/nn/tensorrtapi.cpp
@@ -252,12 +252,10 @@ ICudaEngine* TensorrtAPI::get_cuda_engine() {
252
info_string("serialize engine:", trtFilePath);
253
// serialized engines are not portable across platforms or TensorRT versions
254
// engines are specific to the exact GPU model they were built on
255
- IHostMemory *serializedModel = engine->serialize();
256
unique_ptr<IHostMemory, samplesCommon::InferDeleter> enginePlan{engine->serialize()};
257
// export engine for future uses
258
// write engine to file
259
write_buffer(enginePlan->data(), enginePlan->size(), trtFilePath);
260
- serializedModel->destroy();
261
}
262
263
return engine;
0 commit comments