• L
    Add lgamma_op kernel and unittest (#32913) · 2d3cbb49
    levi131 提交于
    * run pre-commit
    
    * use HOST or DEVICE instead of HOSTDEVICE in implementation of lgamma op
    
    * add test for fp32
    
    * add lgamma to op_threshold_white_list
    
    * add cuda kernel for lgamma kernel
    
    * modify numeric grad delta
    
    * fix small English issue
    
    * change LaunchElementwiseCudaKernel to LaunchSameDimsElementwiseCudaKernel
    2d3cbb49
lgamma_op.cc 3.3 KB