• Y
    [AMP] Add check_numerics API. (#54301) · a5444592
    Yiqun Liu 提交于
    * Add outputs to check_numerics_kernel.
    
    * Add check_numerics to yaml.
    
    * Add API and unittest.
    
    * Add check_nan_inf_level as argument of check_numerics_kernel.
    
    * Add more unittests.
    
    * Fix static API implementation and unittest.
    
    * Move the implementation of check_numerics to paddle.amp.
    
    * Fix import error.
    a5444592
pybind.cc 105.5 KB