From 44c4da94369c1ab1a660b90a70202810337f3cf8 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Shubham=20Gupta/SNAP=20/SRI-Bangalore/Engineer/=EC=82=BC?= =?utf8?q?=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Wed, 2 Jan 2019 07:51:11 +0530 Subject: [PATCH] Add missing condition in Batch_to_space kernel for #if defined (#4101) This patch will add one more condition to check if defined(Batch_OUT). Signed-off-by: shubham --- libs/ARMComputeEx/src/core/CL/cl_kernels/batch_to_space_nd.cl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/ARMComputeEx/src/core/CL/cl_kernels/batch_to_space_nd.cl b/libs/ARMComputeEx/src/core/CL/cl_kernels/batch_to_space_nd.cl index 66da326..0086e39 100644 --- a/libs/ARMComputeEx/src/core/CL/cl_kernels/batch_to_space_nd.cl +++ b/libs/ARMComputeEx/src/core/CL/cl_kernels/batch_to_space_nd.cl @@ -16,7 +16,7 @@ */ #include "helpers.h" -#if defined(DATA_TYPE) && defined(DEPTH_OUT) && defined(BLOCK_SIZE0) && defined(BLOCK_SIZE1) +#if defined(DATA_TYPE) && defined(DEPTH_OUT) && defined(BLOCK_SIZE0) && defined(BLOCK_SIZE1) && defined(BATCH_OUT) /** Perform batch to space rearrangement of tensor * * @attention Data type can be passed using the -DDATA_TYPE compile flag, e.g. -DDATA_TYPE=float @@ -67,4 +67,4 @@ __kernel void batch_to_space_nd( *((__global DATA_TYPE *)out.ptr) = *((__global DATA_TYPE *)tensor4D_offset(&in, in_index[0], in_index[1], in_index[2], in_index[3])); } -#endif // defined(DATA_TYPE) && defined(DEPTH_IN) && defined(BLOCK_SIZE0) && defined(BLOCK_SIZE1) +#endif // defined(DATA_TYPE) && defined(DEPTH_IN) && defined(BLOCK_SIZE0) && defined(BLOCK_SIZE1) && defined(BATCH_OUT) -- 2.7.4