• Pedro Cuenca's avatar
    Run `torch.compile` tests in separate subprocesses (#3503) · bde2cb5d
    Pedro Cuenca authored
    * Run ControlNet compile test in a separate subprocess
    
    `torch.compile()` spawns several subprocesses and the GPU memory used
    was not reclaimed after the test ran. This approach was taken from
    `transformers`.
    
    * Style
    
    * Prepare a couple more compile tests to run in subprocess.
    
    * Use require_torch_2 decorator.
    
    * Test inpaint_compile in subprocess.
    
    * Run img2img compile test in subprocess.
    
    * Run stable diffusion compile test in subprocess.
    
    * style
    
    * Temporarily trigger on pr to test.
    
    * Revert "Temporarily trigger on pr to test."
    
    This reverts commit 82d76868ddf9cc634a9f14b2b0aef1d5433cd750.
    bde2cb5d
testing_utils.py 20.1 KB