• Sudhakar Singh's avatar
    Add unit tests for Fused NovoGrad (#1065) · 59d2f7ac
    Sudhakar Singh authored
    * Add unit tests for fused-novograd
    
    * Fix: tensors should reside on the same device
    
    * Fix: Cudastream should be called on the same device on which the tensors reside on. Found this during debugging fused novograd multi-device unit test
    
    * fixed issues mentioned in the comments
    59d2f7ac
test_fused_optimizer.py 11 KB