Skip to content

Commit dcb50d5

Browse files
committed
hardcode CPU
1 parent e529764 commit dcb50d5

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/torchcodec/decoders/_core/VideoDecoder.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -906,8 +906,7 @@ void VideoDecoder::convertAVFrameToDecodedOutputOnCPU(
906906
"x3, got ",
907907
shape);
908908
} else {
909-
tensor = allocateEmptyHWCTensor(
910-
height, width, streamInfo.options.device.type());
909+
tensor = allocateEmptyHWCTensor(height, width, torch::kCPU);
911910
}
912911
rawOutput.data = tensor.data_ptr<uint8_t>();
913912
convertFrameToBufferUsingSwsScale(rawOutput);

0 commit comments

Comments
 (0)