add a inner loop for index_select_grad_init() in index_select op when dealing...
add a inner loop for index_select_grad_init() in index_select op when dealing with large-shape data (PaddlePaddle#41563) (#41669)
显示
- paddle/phi/kernels/funcs/gather.cu.h 5 个添加, 10 个删除paddle/phi/kernels/funcs/gather.cu.h
- paddle/phi/kernels/funcs/scatter.cu.h 7 个添加, 9 个删除paddle/phi/kernels/funcs/scatter.cu.h
- paddle/phi/kernels/gpu/index_select_grad_kernel.cu 4 个添加, 12 个删除paddle/phi/kernels/gpu/index_select_grad_kernel.cu
- paddle/phi/kernels/gpu/index_select_kernel.cu 1 个添加, 1 个删除paddle/phi/kernels/gpu/index_select_kernel.cu
加载中
想要评论请 注册 或 登录