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 fcbe4a3 commit 7322f9eCopy full SHA for 7322f9e
src/torchcodec/decoders/_core/CudaDevice.cpp
@@ -214,6 +214,7 @@ void convertAVFrameToDecodedOutputOnCuda(
214
preAllocatedOutputTensor.stride(0),
215
oSizeROI);
216
TORCH_CHECK(status == NPP_SUCCESS, "Failed to convert NV12 frame.");
217
+ output.frame = preAllocatedOutputTensor;
218
// Make the pytorch stream wait for the npp kernel to finish before using the
219
// output.
220
at::cuda::CUDAEvent nppDoneEvent;
0 commit comments