• Y
    Add bfloat16 support for several operators and apis. (#52696) · ba9a22db
    Yiqun Liu 提交于
    * Cherry-pick the register of bfloat16 for amp_kernel, pull request #45541.
    
    * Cherry-pick the master_grad support of adamw, pull request #51141.
    
    * add bf16 for some ops in static mode (#51582)
    
    * Add bfloat16 support for some api in static mode.
    
    * Fix codestyle.
    
    * Revert the change of layer_function_generator.py.
    
    ---------
    Co-authored-by: shaojie_wang's avatarShaojie WANG <wsjmessi@163.com>
    ba9a22db
layer_function_generator.py 13.7 KB