"googlemock/include/vscode:/vscode.git/clone" did not exist on "490554aa0f3618e1e5dd217f11fe0c3f188ed615"
  • Hongxin Liu's avatar
    [bf16] add bf16 support (#3882) · ae02d4e4
    Hongxin Liu authored
    * [bf16] add bf16 support for fused adam (#3844)
    
    * [bf16] fused adam kernel support bf16
    
    * [test] update fused adam kernel test
    
    * [test] update fused adam test
    
    * [bf16] cpu adam and hybrid adam optimizers support bf16 (#3860)
    
    * [bf16] implement mixed precision mixin and add bf16 support for low level zero (#3869)
    
    * [bf16] add mixed precision mixin
    
    * [bf16] low level zero optim support bf16
    
    * [text] update low level zero test
    
    * [text] fix low level zero grad acc test
    
    * [bf16] add bf16 support for gemini (#3872)
    
    * [bf16] gemini support bf16
    
    * [test] update gemini bf16 test
    
    * [doc] update gemini docstring
    
    * [bf16] add bf16 support for plugins (#3877)
    
    * [bf16] add bf16 support for legacy zero (#3879)
    
    * [zero] init context support bf16
    
    * [zero] legacy zero support bf16
    
    * [test] add zero bf16 test
    
    * [doc] add bf16 related docstring for legacy zero
    ae02d4e4
test_adam_optim.py 3.27 KB