From 1ab03d49d5161dfeb5c8251d7609b78e3ff29440 Mon Sep 17 00:00:00 2001 From: guosheng Date: Wed, 29 Nov 2017 13:47:19 +0800 Subject: [PATCH] Fix gru_op related code style in gpu_kernel --- paddle/operators/math/detail/gru_gpu_kernel.h | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/paddle/operators/math/detail/gru_gpu_kernel.h b/paddle/operators/math/detail/gru_gpu_kernel.h index f3983c51958..d2edcb7f258 100644 --- a/paddle/operators/math/detail/gru_gpu_kernel.h +++ b/paddle/operators/math/detail/gru_gpu_kernel.h @@ -36,12 +36,12 @@ __global__ void KeGruForwardResetOutput(OpResetOutput op_reset_output, T *prev_output_value, int frame_size, int batch_size, activation_mode_t active_gate) { - const int frame_idx = block_idx.x * block_dim.x + thread_idx.x; + const int frame_idx = blockIdx.x * blockDim.x + threadIdx.x; if (frame_idx >= frame_size) return; int batch_idx = 0; if (is_batch) { - batch_idx = block_idx.y * block_dim.y + thread_idx.y; + batch_idx = blockIdx.y * blockDim.y + threadIdx.y; if (batch_idx >= batch_size) return; gate_value += batch_idx * 3 * frame_size; reset_output_value += batch_idx * frame_size; @@ -75,11 +75,11 @@ __global__ void KeGruForwardFinalOutput(OpFinalOutput op_final_output, T *output_value, int frame_size, int batch_size, activation_mode_t active_node) { - const int frame_idx = block_idx.x * block_dim.x + thread_idx.x; + const int frame_idx = blockIdx.x * blockDim.x + threadIdx.x; if (frame_idx >= frame_size) return; int batch_idx = 0; if (is_batch) { - batch_idx = block_idx.y * block_dim.y + thread_idx.y; + batch_idx = blockIdx.y * blockDim.y + threadIdx.y; if (batch_idx >= batch_size) return; gate_value += batch_idx * 3 * frame_size; output_value += batch_idx * frame_size; @@ -112,11 +112,11 @@ __global__ void KeGruBackwardStateGrad(OpStateGrad op_state_grad, T *gate_value, T *prev_out_grad, T *output_grad, int frame_size, int batch_size, activation_mode_t active_node) { - const int frame_idx = block_idx.x * block_dim.x + thread_idx.x; + const int frame_idx = blockIdx.x * blockDim.x + threadIdx.x; if (frame_idx >= frame_size) return; int batch_idx = 0; if (is_batch) { - batch_idx = block_idx.y * block_dim.y + thread_idx.y; + batch_idx = blockIdx.y * blockDim.y + threadIdx.y; if (batch_idx >= batch_size) return; gate_value += batch_idx * 3 * frame_size; gate_grad += batch_idx * 3 * frame_size; @@ -160,11 +160,11 @@ __global__ void KeGruBackwardResetGrad(OpResetGrad op_reset_grad, T *gate_value, T *prev_out_grad, T *reset_output_grad, int frame_size, int batch_size, activation_mode_t active_gate) { - const int frame_idx = block_idx.x * block_dim.x + thread_idx.x; + const int frame_idx = blockIdx.x * blockDim.x + threadIdx.x; if (frame_idx >= frame_size) return; int batch_idx = 0; if (is_batch) { - batch_idx = block_idx.y * block_dim.y + thread_idx.y; + batch_idx = blockIdx.y * blockDim.y + threadIdx.y; if (batch_idx >= batch_size) return; gate_value += batch_idx * 3 * frame_size; gate_grad += batch_idx * 3 * frame_size; -- GitLab