• C
    [Phi] Move amp ops into phi (#45079) · b4f67757
    Chen Weihang 提交于
    * move check finite and unscale kernel into phi
    
    * move infershape into phi
    
    * move update_loss_scaling kernel into phi
    
    * remove original kernels
    
    * move update loss scaling infershape into phi
    
    * add header for xpu and npu
    
    * solve coverage failed
    
    * fix npu test failed
    
    * remove mutable data in cu file
    
    * fix new executor failed
    
    * add valid check for meta tensor output
    b4f67757
update_loss_scaling_sig.cc 1.6 KB