1. 08 3月, 2022 1 次提交
  2. 18 2月, 2022 1 次提交
  3. 16 2月, 2022 1 次提交
    • L
      [bf16] pten matmul cuda kernel support bf16 (#39485) · d5a0d31a
      Leo Chen 提交于
      * pten matmul cuda kernel support bf16
      
      * fix pten kernel name
      
      * add matmul_grad bf16 kernel
      
      * add emptylike bf16 kernel
      
      * fix compile
      
      * suppport rocm
      
      * fix error
      
      * fix rocm
      
      * add bf16 header file
      
      * fix compile
      d5a0d31a
  4. 10 2月, 2022 1 次提交
  5. 13 1月, 2022 2 次提交
  6. 04 1月, 2022 1 次提交
  7. 17 12月, 2021 1 次提交
  8. 02 12月, 2021 1 次提交
  9. 25 11月, 2021 2 次提交
  10. 23 11月, 2021 1 次提交
  11. 12 11月, 2021 1 次提交
  12. 04 11月, 2021 1 次提交
  13. 01 11月, 2021 1 次提交
  14. 28 10月, 2021 1 次提交
  15. 26 10月, 2021 1 次提交
  16. 25 10月, 2021 1 次提交
  17. 21 10月, 2021 3 次提交
  18. 20 10月, 2021 1 次提交
  19. 19 10月, 2021 1 次提交
  20. 18 10月, 2021 2 次提交
  21. 14 10月, 2021 1 次提交
  22. 12 10月, 2021 3 次提交
  23. 11 10月, 2021 2 次提交
  24. 08 10月, 2021 1 次提交
  25. 30 9月, 2021 1 次提交
  26. 29 9月, 2021 4 次提交
  27. 22 9月, 2021 1 次提交
  28. 16 9月, 2021 1 次提交
  29. 15 9月, 2021 1 次提交
    • P
      [NPU] add beam_search npu op (#34860) · 3760be06
      pangyoki 提交于
      * add beam_search npu op
      
      * fix CMakeList and add unittest
      
      * fix bug of beam search npu op
      
      * fix unittest
      
      * let input ids become int64
      
      * set output ids to int64_t
      
      * delete check_dygraph
      
      * fix beam_width=1
      3760be06