From db00bc9276a75000f26e67f20178484b5b08e87d Mon Sep 17 00:00:00 2001
From: yangfei <yangfei11@baidu.com>
Date: Tue, 16 Oct 2018 16:05:15 +0800
Subject: [PATCH] imp add op kernel

---
 src/operators/kernel/cl/elementwise_add_kernel.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/operators/kernel/cl/elementwise_add_kernel.cpp b/src/operators/kernel/cl/elementwise_add_kernel.cpp
index 6cc7b819b1..a64f3c688b 100644
--- a/src/operators/kernel/cl/elementwise_add_kernel.cpp
+++ b/src/operators/kernel/cl/elementwise_add_kernel.cpp
@@ -64,7 +64,7 @@ void ElementwiseAddKernel<GPU_CL, float>::Compute(
     cl_mem input_image = input->GetCLImage();
     cl_mem bias_image = bias->GetCLImage();
     cl_mem output_image = output->GetCLImage();
-    int tensor_w = input->dims()[4];
+    int tensor_w = input->dims()[3];
     status = clSetKernelArg(kernel, 0, sizeof(cl_mem), (void *)&input_image);
     CL_CHECK_ERRORS(status);
     status = clSetKernelArg(kernel, 1, sizeof(cl_mem), (void *)&bias_image);
-- 
GitLab