File tree Expand file tree Collapse file tree 2 files changed +4
-1
lines changed
Expand file tree Collapse file tree 2 files changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -342,6 +342,9 @@ std::optional<const AVCodec*> CudaDeviceInterface::findCodec(
342342 void * i = nullptr ;
343343 const AVCodec* codec = nullptr ;
344344 while ((codec = av_codec_iterate (&i)) != nullptr ) {
345+ TORCH_CHECK (
346+ codec != nullptr ,
347+ " codec returned by av_codec_iterate should not be null" );
345348 if (isDecoder) {
346349 if (codec->id != codecId || !av_codec_is_decoder (codec)) {
347350 continue ;
Original file line number Diff line number Diff line change @@ -160,7 +160,7 @@ class DeviceInterface {
160160 virtual std::optional<const AVCodec*> findHardwareEncoder (
161161 [[maybe_unused]] const AVCodecID& codecId) {
162162 TORCH_CHECK (false );
163- };
163+ }
164164
165165 protected:
166166 torch::Device device_;
You can’t perform that action at this time.
0 commit comments