1. 04 3月, 2020 1 次提交
    • Z
      Add flags to limit gpu memory (#22793) · d41d802b
      Zeng Jinle 提交于
      * add recorded cuda memory apis, fix typo, test=develop
      
      * add more ut, test=develop
      
      * follow comments, test=develop
      
      * fix py35 incompatible issues, test=develop
      d41d802b
  2. 09 1月, 2020 1 次提交
  3. 08 1月, 2020 1 次提交
  4. 14 11月, 2019 1 次提交
  5. 06 11月, 2019 1 次提交
  6. 01 11月, 2019 1 次提交
  7. 12 10月, 2019 1 次提交
  8. 16 8月, 2019 1 次提交
  9. 01 8月, 2019 1 次提交
  10. 31 7月, 2019 1 次提交
  11. 18 7月, 2019 1 次提交
  12. 16 7月, 2019 1 次提交
  13. 30 4月, 2019 1 次提交
  14. 21 3月, 2019 1 次提交
    • S
      add more unittest · 953214ad
      sneaxiy 提交于
      modify allocator strategy
      remove changes of legacy buddy_allocator
      test=develop
      953214ad
  15. 19 3月, 2019 1 次提交
  16. 24 1月, 2019 2 次提交
    • Y
      Add the CUDA kernel for beam_search op (#15020) · 3008fa12
      Yiqun Liu 提交于
      * Refine the beam_search op and test.
      
      * A basic CUDA implementation of beam_search for small batch_size.
      
      * Implement CUDA kernel for beam_search_op.
      
      * Use multiple CUDA threads in the same block to select the top beam.
      
      * Update the python api of beam_search op.
      
      * Enable extend function in CPU kernel of beam_search op.
      
      * Unify the CUDA codes.
      test=develop
      
      * Unify the CPU kernel of beam_search op.
      
      * Ensure the seletced items of beam_search_op's CPU kernel sorted by scores.
      
      * Update the description of beam_search in API.spec.
      
      * Enable the use of CUDA kernel in beam_search op.
      
      * Exclude the beam_search's CUDA unittest when there is no CUDA gpu, and delete some debuging statements.
      test=develop
      
      * Follow comments.
      test=develop
      
      * Call the CPU kernel for beam_search op when batch_size > 4.
      test=develop
      
      * Remove the except of is_empty op in PrepareData.
      test=develop
      3008fa12
    • S
      lazy_allocator · 51227bd4
      sneaxiy 提交于
      test=develop
      51227bd4
  17. 04 12月, 2018 1 次提交
  18. 27 11月, 2018 1 次提交
  19. 26 11月, 2018 2 次提交
  20. 22 11月, 2018 2 次提交
    • C
      Refine cublas to support CUBLAS_TENSOR_OP_MATH (#13929) · 00b9e9a1
      chengduo 提交于
      * refine cublase
      test=develop
      
      * code refine
      
      * refine cublas
      
      * add GEMME_EX
      
      * add enable_cublas_tensor_op_math doc and add cublasCall
      test=develop
      
      * fix CublasCall for cuda version
      test=develop
      
      * fix error
      test=develop
      
      * fix GEMM_EX to be compatible with gcc 4.8
      test=develop
      
      * add GEMM_EX
      test=develop
      
      * to compatiable with gcc4.8
      test=develop
      00b9e9a1
    • P
      fix unit test cases · 7c8c9dc9
      peizhilin 提交于
      7c8c9dc9
  21. 08 11月, 2018 1 次提交
  22. 15 10月, 2018 1 次提交
  23. 08 10月, 2018 1 次提交
  24. 27 9月, 2018 1 次提交
  25. 26 9月, 2018 1 次提交
  26. 14 8月, 2018 1 次提交
  27. 08 8月, 2018 1 次提交
  28. 23 4月, 2018 1 次提交
  29. 08 4月, 2018 1 次提交
  30. 10 3月, 2018 1 次提交
  31. 03 3月, 2018 1 次提交
  32. 12 2月, 2018 1 次提交
  33. 10 2月, 2018 2 次提交
  34. 10 1月, 2018 2 次提交
  35. 22 12月, 2017 1 次提交
    • D
      "remove GPU Sync Interface" (#6793) · abde3130
      dzhwinter 提交于
      * "remove GPU Sync Interface"
      
      * "fix typo"
      
      * "fix type cast error"
      
      * "fix related Copy with stream"
      
      * "fix failed tests with DevicePool"
      
      * "fix stupid removed position error"
      abde3130