1. 13 9月, 2021 2 次提交
  2. 28 5月, 2021 1 次提交
    • 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