1. 26 6月, 2022 1 次提交
  2. 15 3月, 2022 1 次提交
    • X
      [phi] Transfer lgamma, kldiv_loss, isclose, cumprod kernels into phi and pass... · 64223620
      xiongkun 提交于
      [phi] Transfer lgamma, kldiv_loss, isclose, cumprod kernels into phi and pass the tests of these four kernels (#39770)
      
      * tranfer and pass the lgamma unittest
      
      * merge and pass the test
      
      * transfer kldiv_loss and kldiv_loss_grad; pass the unitest
      
      * trafer the isclose and cumprod kernel
      
      * change PT_REGISTER -> PD_REGISTER
      
      * fix by code review
      
      * fix by code review
      
      * fix
      
      * remove enforce include dependence from scalar
      
      * fix
      
      * fix by code review
      
      * fix by code review
      64223620
  3. 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