You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Yiqun Liu 3008fa1261 Add the CUDA kernel for beam_search op (#15020) ... * 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		6 years ago
..
paddle	Add the CUDA kernel for beam_search op (#15020)	6 years ago
.gitignore	Add gitignores	7 years ago
CMakeLists.txt	remove legacy WITH_SWIG_PY option	6 years ago
__init__.py	Firt timeline version	7 years ago
requirements.txt	Init paddle slim (#14834)	6 years ago
setup.py.in	Merge branch 'develop' of https://github.com/PaddlePaddle/Paddle into graph_quantization	6 years ago