9 Commits (45702951226401a24df501960d7fd9b47152083d)

Author	SHA1	Message	Date
zhaoyuchen2018	792443ef23	Refine elementwise kernel. (#16952) ... * Refine elementwise kernel. Add a simple cuda kernel if grad x and y both exist Use 2D block cuda kernel to do broadcast. test=develop Signed-off-by: zhaoyuchen <zhaoyuchen01@baidu.com> * refine code. test=develop Signed-off-by: zhaoyuchen <zhaoyuchen01@baidu.com> * refine code. test=develop Signed-off-by: zhaoyuchen <zhaoyuchen01@baidu.com>	6 years ago
guoshengCS	5dfce93101	To make CUDA_LAUNCH_KERNEL_HELPER support large size. ... test=develop	6 years ago
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
dzhwinter	2673798ddb	"fix float16 ShuffleDownSync Bug" (#12756) ... * "fix bug" * "add test case"	7 years ago
dzhwinter	39ac9e39c2	float16 type support enhance (#12181) ... * cherry picked * "cherry picked platform" * "add comment" * "fix ci"	7 years ago
chengduoZH	345737d0fe	add sync	7 years ago
chengduoZH	d36af62c1e	wrap_shfl_x_sync	7 years ago
chengduoZH	e97c1a8ca0	fix __shfl	7 years ago
chengduo	4fbde42cdf	Fix __shfl_down_sync_ of cross_entropy (#10345) ... * fix __shfl_down_sync_ of cross_entropy * use reduceSum * "fix ci"	7 years ago