Skip to content

Commit 48e88bf

Browse files
author
Daniel Flores
committed
static cast ints
1 parent 1f3ebc7 commit 48e88bf

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/torchcodec/_core/Encoder.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -598,8 +598,8 @@ void VideoEncoder::initializeEncoder(
598598
// TODO-VideoEncoder: Remove assumption that tensor in NCHW format
599599
auto sizes = frames_.sizes();
600600
inPixelFormat_ = AV_PIX_FMT_GBRP;
601-
inHeight_ = sizes[2];
602-
inWidth_ = sizes[3];
601+
inHeight_ = static_cast<int>(sizes[2]);
602+
inWidth_ = static_cast<int>(sizes[3]);
603603

604604
// Use specified dimensions or input dimensions
605605
// TODO-VideoEncoder: Allow height and width to be set
@@ -664,7 +664,7 @@ void VideoEncoder::encode() {
664664
getFFMPEGErrorStringFromErrorCode(status));
665665

666666
AutoAVPacket autoAVPacket;
667-
int numFrames = frames_.sizes()[0];
667+
int numFrames = static_cast<int>(frames_.sizes()[0]);
668668
for (int i = 0; i < numFrames; ++i) {
669669
torch::Tensor currFrame = frames_[i];
670670
UniqueAVFrame avFrame = convertTensorToAVFrame(currFrame, i);

0 commit comments

Comments
 (0)