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 33ae295 commit 001bbc8Copy full SHA for 001bbc8
src/torchcodec/_core/BetaCudaDeviceInterface.cpp
@@ -111,7 +111,7 @@ static UniqueCUvideodecoder createDecoder(CUVIDEOFORMAT* videoFormat) {
111
// Below we'll set the decoderParams.OutputFormat to NV12, so we need to make
112
// sure it's actually supported.
113
TORCH_CHECK(
114
- caps.nOutputFormatMask & (1 << cudaVideoSurfaceFormat_NV12),
+ (caps.nOutputFormatMask >> cudaVideoSurfaceFormat_NV12) & 1,
115
"NV12 output format is not supported for this configuration. ",
116
"Codec: ",
117
static_cast<int>(videoFormat->codec),
0 commit comments