1. 18 7月, 2019 1 次提交
    • Z
      Feature/auto_growth_allocator (#18561) · ae58afc5
      Zeng Jinle 提交于
      * feature/auto_growth_allocator, test=develop
      
      * add unittest of AlignedAllocator, test=develop
      
      * try to turn on auto_growth to test on CI, test=develop
      
      * fix segmentation fault in mixed_vector.h, test=develop
      
      * add unittests, test=develop
      ae58afc5
  2. 24 1月, 2019 1 次提交
    • 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
  3. 26 12月, 2018 1 次提交
  4. 18 12月, 2018 2 次提交
  5. 27 11月, 2018 1 次提交
  6. 21 11月, 2018 1 次提交
  7. 15 11月, 2018 1 次提交
  8. 17 10月, 2018 1 次提交
  9. 28 9月, 2018 1 次提交
  10. 21 9月, 2018 1 次提交
  11. 20 9月, 2018 2 次提交
  12. 17 9月, 2018 3 次提交
  13. 25 7月, 2018 1 次提交
  14. 24 7月, 2018 1 次提交
  15. 23 7月, 2018 1 次提交
  16. 19 4月, 2018 1 次提交
  17. 22 3月, 2018 1 次提交
  18. 15 2月, 2018 1 次提交
    • Y
      Update tensor_util.h (#8422) · cfffb1a3
      Yi Wang 提交于
      * Update tensor_util.h
      
      * Update with moved TensorDesc
      
      * Fix tensur_utils.cu
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Make tensor_util.cu a symbolic link
      cfffb1a3
  19. 13 2月, 2018 1 次提交
  20. 12 2月, 2018 1 次提交
  21. 11 2月, 2018 8 次提交
  22. 10 2月, 2018 2 次提交
  23. 09 2月, 2018 3 次提交
  24. 08 2月, 2018 1 次提交
  25. 06 2月, 2018 2 次提交