From 43f1358fafccedb26bc73d2c6c1e373c10bba6bb Mon Sep 17 00:00:00 2001 From: Pei Yang Date: Wed, 20 Nov 2019 23:21:25 +0800 Subject: [PATCH] fix sequence pool cuda (#2466) * add sequence_pool cuda kernel, test=develop * fix sequence_pool cuda,test=develop * fix and complete unittest, test=develop * fix macro of sequence_pool cuda, test=develop --- lite/kernels/cuda/sequence_pool_compute.cu | 13 ------------- 1 file changed, 13 deletions(-) diff --git a/lite/kernels/cuda/sequence_pool_compute.cu b/lite/kernels/cuda/sequence_pool_compute.cu index 0b0376859f..57b69f7a7a 100644 --- a/lite/kernels/cuda/sequence_pool_compute.cu +++ b/lite/kernels/cuda/sequence_pool_compute.cu @@ -17,19 +17,6 @@ #include "lite/core/target_wrapper.h" #include "lite/kernels/cuda/sequence_pool_compute.h" -const int CUDA_NUM_THREADS = 512; -#define CUDA_KERNEL_LOOP(i, n) \ - for (int i = blockIdx.x * blockDim.x + threadIdx.x; i < (n); \ - i += blockDim.x * gridDim.x) - -/// CUDA: number of blocks for threads. -inline int CUDA_GET_BLOCKS(const int N) { - return (N + CUDA_NUM_THREADS - 1) / CUDA_NUM_THREADS; -} -inline int CUDA_GET_BLOCKS(const int N, const int base) { - return (N + base - 1) / base; -} - namespace paddle { namespace lite { namespace kernels { -- GitLab