• C
    [Phi] Move softmax with cross entropy kernel into phi (#40832) · e6ec98fe
    Chen Weihang 提交于
    * add cross_entropy_with_softmax phi kernel
    
    * remove softmax_with_cross_entropy kernel
    
    * add softmax_with_cross_entropy grad kernel
    
    * remove original op kernel
    
    * refine cross entropy impl
    
    * fix pointer error
    
    * revert kernel cu change
    
    * fix xpu failed
    
    * fix cinn failed
    
    * fix npu failed
    
    * add forward sig
    
    * add check_nan_inf for pt kernel
    
    * remove repeat cmake item
    
    * fix unittest error
    e6ec98fe
softmax_with_cross_entropy_sig.cc 2.1 KB