| 409 | } |
| 410 | |
| 411 | void hostToDevice(TrtCudaStream& stream) override |
| 412 | { |
| 413 | cudaCheck(cudaMemcpyAsync(mDeviceBuffer.get(), mHostBuffer.get(), mSize, cudaMemcpyHostToDevice, stream.get())); |
| 414 | } |
| 415 | |
| 416 | void deviceToHost(TrtCudaStream& stream) override |
| 417 | { |