Skip to content

Commit 09fb932

Browse files
authored
More tests to MultiStreamEncoder (#1408)
1 parent da5b1cb commit 09fb932

2 files changed

Lines changed: 412 additions & 7 deletions

File tree

src/torchcodec/_core/Encoder.cpp

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1153,6 +1153,13 @@ int MultiStreamEncoder::addAudioStream(
11531153
STD_TORCH_CHECK(sampleRate > 0, "sample_rate must be > 0, got ", sampleRate);
11541154
STD_TORCH_CHECK(
11551155
numChannels > 0, "num_channels must be > 0, got ", numChannels);
1156+
STD_TORCH_CHECK(
1157+
numChannels <= AV_NUM_DATA_POINTERS,
1158+
"Trying to encode ",
1159+
numChannels,
1160+
" channels, but FFmpeg only supports ",
1161+
AV_NUM_DATA_POINTERS,
1162+
" channels per frame.");
11561163

11571164
AudioStream audioStream;
11581165
audioStream.inSampleRate = sampleRate;

0 commit comments

Comments
 (0)