• Nayan Singhal's avatar
    Add Unit test cases for BMUF · b5f41f82
    Nayan Singhal authored
    Summary:
    This unit test guards the bmuf code.
    
    change:
    1. distributed_init assumes we are always using cuda device which is not the case if you are using "gloo" backend on CPU machine.
    
    Reviewed By: jay-mahadeokar
    
    Differential Revision: D17821391
    
    fbshipit-source-id: 28e1bb39f7a4889b1dc6bd636b7c499e55bfc69a
    b5f41f82
test_bmuf.py 4.22 KB