• Benjamin Lefaudeux's avatar
    [fix] Flaky tests (#283) · 79365ee6
    Benjamin Lefaudeux authored
    * adding the pytest timeout plugin to properly root out hanging tests
    * removing redundant code, slightly more reasonable timeout, works on single cuda
    * finding the root bug for some of the cpu hangs, rpc init
    * propagating all the rpc init test changes to the pipe and model parallel tests
    79365ee6
test_sharded_ddp.py 16.9 KB