Skip to content

Commit ee3a199

Browse files
committed
Add assertion
1 parent f49d507 commit ee3a199

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/torchcodec/_core/Encoder.cpp

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -249,6 +249,14 @@ void AudioEncoder::encodeInnerLoop(
249249
avCodecContext_->sample_fmt,
250250
srcAVFrame->sample_rate, // No sample rate conversion
251251
srcAVFrame->sample_rate);
252+
TORCH_CHECK(
253+
convertedAVFrame->nb_samples == srcAVFrame->nb_samples,
254+
"convertedAVFrame->nb_samples=",
255+
convertedAVFrame->nb_samples,
256+
" differs from ",
257+
"srcAVFrame->nb_samples=",
258+
srcAVFrame->nb_samples,
259+
"This is unexpected, please report on the TorchCodec bug tracker.");
252260
}
253261
const UniqueAVFrame& avFrame = mustConvert ? convertedAVFrame : srcAVFrame;
254262

0 commit comments

Comments
 (0)