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 e529764 commit dcb50d5Copy full SHA for dcb50d5
src/torchcodec/decoders/_core/VideoDecoder.cpp
@@ -906,8 +906,7 @@ void VideoDecoder::convertAVFrameToDecodedOutputOnCPU(
906
"x3, got ",
907
shape);
908
} else {
909
- tensor = allocateEmptyHWCTensor(
910
- height, width, streamInfo.options.device.type());
+ tensor = allocateEmptyHWCTensor(height, width, torch::kCPU);
911
}
912
rawOutput.data = tensor.data_ptr<uint8_t>();
913
convertFrameToBufferUsingSwsScale(rawOutput);
0 commit comments