| 247 | } |
| 248 | |
| 249 | void TensorRTRuntimeOpr::LoadDumpImpl::dump( |
| 250 | serialization::OprDumpContext& ctx, const cg::OperatorNodeBase& opr) { |
| 251 | TensorRTUniquePtr<nvinfer1::IHostMemory> buf{ |
| 252 | opr.cast_final_safe<Opr>().trt_cuda_engine()->serialize(), {}}; |
| 253 | mgb_assert(buf, "failed to serialize ICudaEngine"); |
| 254 | ctx.dump_buf_with_len(buf->data(), buf->size()); |
| 255 | } |
| 256 | |
| 257 | cg::OperatorNodeBase* TensorRTRuntimeOpr::LoadDumpImpl::load( |
| 258 | serialization::OprLoadContext& ctx, const cg::VarNodeArray& inputs, |
nothing calls this directly
no test coverage detected