| 190 | } |
| 191 | |
| 192 | void endCapture(TrtCudaStream& stream) |
| 193 | { |
| 194 | cudaCheck(cudaStreamEndCapture(stream.get(), &mGraph)); |
| 195 | cudaCheck(cudaGraphInstantiate(&mGraphExec, mGraph, nullptr, nullptr, 0)); |
| 196 | cudaCheck(cudaGraphDestroy(mGraph)); |
| 197 | } |
| 198 | |
| 199 | void endCaptureOnError(TrtCudaStream& stream) |
| 200 | { |
no test coverage detected