Fix GPU memory leak on StreamReader (#3186)
Summary: Pull Request resolved: https://github.com/pytorch/audio/pull/3186 Fix the GPU memory leak introduced in https://github.com/pytorch/audio/pull/3183 The HW frames context is owned by AVCodecContext. The removed `av_buffer_ref` call increased the ferenrence counting unnecessarily, and prevented AVCodecContext from feeing the resource. (Note: this ignores all push blocking failures!) Reviewed By: nateanl Differential Revision: D44231876 fbshipit-source-id: 9be2c33049dd02a3fa82a85271de7fb62e5b09ea
Showing
Please register or sign in to comment