• Y
    [AMP] Unify the static amp codes of fp16 and bf16. (#52694) · dfcba7f4
    Yiqun Liu 提交于
    * Unify the static amp codes of fp16 and bf16.
    
    * Polish apis and add unittest.
    
    * Add operator stats collecting tools for program.
    
    * Add the check of number of bloat16 operators in unittest.
    
    * Add warning for operator not supported for amp.
    
    * Add testing of BF16 O1 and O2.
    dfcba7f4
amp_base_models.py 6.3 KB