Skip to content

Commit 7322f9e

Browse files
committed
Fix CUDA
1 parent fcbe4a3 commit 7322f9e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/torchcodec/decoders/_core/CudaDevice.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -214,6 +214,7 @@ void convertAVFrameToDecodedOutputOnCuda(
214214
preAllocatedOutputTensor.stride(0),
215215
oSizeROI);
216216
TORCH_CHECK(status == NPP_SUCCESS, "Failed to convert NV12 frame.");
217+
output.frame = preAllocatedOutputTensor;
217218
// Make the pytorch stream wait for the npp kernel to finish before using the
218219
// output.
219220
at::cuda::CUDAEvent nppDoneEvent;

0 commit comments

Comments
 (0)