1. 24 8月, 2022 1 次提交
    • L
      make tensor_util contains no cuda code (#45256) · 78916a7a
      Leo Chen 提交于
      * make tensor_util contains no cuda code
      
      * refine isfinite
      
      * revert ut
      
      * move isfinite function to its op
      
      * fix test
      
      * fix compile
      
      * std::isnan is not defined for int type on windows
      
      * fix windows compile
      
      * fix fp16
      
      * fix rocm compile
      
      * revert gradient node
      78916a7a
  2. 16 8月, 2022 1 次提交
    • 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