• L
    [NPU] refactor check_finite_and_scale npu kernel (#32407) · 39a59dcf
    Leo Chen 提交于
    * refactor_check_finite_and_scale_npu_kernel
    
    * fix compile
    
    * add alloc_float_status op
    
    * add alloc_float_status op
    
    * add FloatStatus for check_finite_and_unscale
    
    * refine code
    
    * remove unneccessary logic
    
    * refine for fleet
    39a59dcf
alloc_float_status_op.cc 2.4 KB