• L
    Feature/Enable Auto-Mixed-Precision in dynamic graph (#24903) · 2d95280e
    Leo Chen 提交于
    * add auto_cast, test=develop
    
    * add loss scaler, test=develop
    
    * add comments, test=develop
    
    * refine code, test=develop
    
    * refine code, test=develop
    
    * do not set flags automatically, test=develop
    
    * fix custom op bug, test=develop
    
    * add more test, test=develop
    
    * refine enable logic, test=develop
    
    * enable amp test with GPU, test=develop
    
    * add unittest
    
    * add test for found_inf
    
    * follow comments
    
    * follow comments
    
    * remove global variable, use singleton
    
    * add some notes
    
    * update comments
    
    * update comments
    
    * update comments
    
    * add use_dynamic_loss_scaling argument
    
    * refine found_inf
    
    * refine found_inf
    2d95280e
auto_cast.py 5.9 KB