Skip to content

Commit 9f357c7

Browse files
committed
Do all nv12 conversions on GPU
1 parent 6668f4b commit 9f357c7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/torchcodec/_core/CudaDeviceInterface.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -254,7 +254,7 @@ UniqueAVFrame CudaDeviceInterface::maybeConvertAVFrameToNV12(
254254
"That's unexpected, please report this to the TorchCodec repo.");
255255
256256
*/
257-
filters << "hwdownload,format=rgb24";
257+
filters << "hwupload,format=nv12";
258258
} else {
259259
// Actual output color format will be set via filter options
260260
outputFormat = AV_PIX_FMT_CUDA;

0 commit comments

Comments
 (0)