1. 18 11月, 2022 1 次提交
  2. 27 7月, 2022 1 次提交
  3. 26 6月, 2022 1 次提交
  4. 23 2月, 2022 1 次提交
  5. 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
  6. 12 1月, 2022 1 次提交
    • L
      Adjust warpper of gpu_lanuch_config (#38654) · f5166284
      limingshu 提交于
      * first commit
      
      * fix wrong filename
      
      * fix the wrong spell name
      
      * fix gpu config warper
      
      * modify according to pr advices
      
      * fix GpuLauchConfig1D api bugs
      
      * change the config for dropout grad
      
      * fix bugs
      
      * modification according to pr advices
      
      * modification according to pr advices
      f5166284
  7. 03 12月, 2021 1 次提交
  8. 02 6月, 2021 1 次提交
  9. 25 5月, 2021 1 次提交
    • C
      modify complex template for elementwise ops (#33071) · dbc08d69
      chentianyu03 提交于
      * modify complex template for elementwise ops
      
      * modify mul, div grad struct
      
      * add complex template for CudaShuffleDownSync CudaShuffleXorSync funcs and fix the bug when delete cuda<9000
      
      * fix shuffle func args bug
      
      * fix shuffle func args bug
      
      * fix shuffle func args bug
      dbc08d69
  10. 31 3月, 2021 1 次提交
  11. 08 2月, 2021 1 次提交
  12. 01 12月, 2020 1 次提交
  13. 08 5月, 2019 1 次提交
  14. 31 1月, 2019 1 次提交
  15. 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
  16. 17 8月, 2018 1 次提交
  17. 30 7月, 2018 1 次提交
  18. 08 5月, 2018 1 次提交
  19. 04 5月, 2018 1 次提交
  20. 03 5月, 2018 2 次提交