• moto's avatar
    Support CUDA frame in FilterGraph (#3183) · c5b96558
    moto authored
    Summary:
    This commit adds CUDA frame support to FilterGraph
    
    It initializes and attaches CUDA frames context to FilterGraph,
    so that CUDA frames can be processed in FilterGraph.
    
    As a result, it enables
    1. CUDA filter support such as `scale_cuda`
    2. Properly retrieve the pixel format coming out of FilterGraph when
       CUDA HW acceleration is enabled. (currently it is reported as "cuda")
    
    Resolves https://github.com/pytorch/audio/issues/3159
    
    Pull Request resolved: https://github.com/pytorch/audio/pull/3183
    
    Reviewed By: hwangjeff
    
    Differential Revision: D44183722
    
    Pulled By: mthrok
    
    fbshipit-source-id: 522d21039c361ddfaa87fa89cf49c19d210ac62f
    c5b96558
unittest-linux-gpu.yml 3.17 KB