1. 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
  2. 26 5月, 2021 6 次提交
  3. 25 5月, 2021 3 次提交
  4. 24 5月, 2021 3 次提交
  5. 22 5月, 2021 1 次提交
    • J
      Added oneDNN matmul grad BF16/FP32 kernel (#32968) · e2a3a6f7
      jakpiase 提交于
      * added support for most matmul cases
      
      * added more functionality
      
      * full functionality of matmul op, fp32 only
      
      * added bf16 tests and functionality
      
      * added formatting
      
      * changes after review
      
      * minor change
      
      * added reviewers suggestions
      e2a3a6f7
  6. 21 5月, 2021 3 次提交
  7. 20 5月, 2021 3 次提交
  8. 19 5月, 2021 4 次提交
  9. 18 5月, 2021 5 次提交
  10. 17 5月, 2021 1 次提交
  11. 14 5月, 2021 3 次提交
  12. 13 5月, 2021 3 次提交
  13. 12 5月, 2021 4 次提交