Fixed a bug in im2col confusing first and second workgroup size; made im2col kernel 2d instead of 3d

1 files changed, 4 insertions, 4 deletions

diff --git a/src/routines/levelx/xim2col.cpp b/src/routines/levelx/xim2col.cpp
index 51171eb5..527695c0 100644
--- a/src/routines/levelx/xim2col.cpp
+++ b/src/routines/levelx/xim2col.cpp
@@ -71,10 +71,10 @@ void Xim2col<T>::DoIm2col(const size_t channels, const size_t height, const size
   kernel.SetArgument(15, static_cast<int>(col_offset));
 
   // Launches the kernel
-  const auto w_ceiled = Ceil(output_w, db_["COPY_DIMY"]);
-  const auto h_ceiled = Ceil(output_h, db_["COPY_DIMX"]);
-  const auto global = std::vector<size_t>{w_ceiled, h_ceiled, channels};
-  const auto local = std::vector<size_t>{db_["COPY_DIMX"], db_["COPY_DIMY"], 1};
+  const auto w_ceiled = Ceil(output_w, db_["COPY_DIMX"]);
+  const auto h_ceiled = Ceil(output_h, db_["COPY_DIMY"]);
+  const auto global = std::vector<size_t>{w_ceiled, h_ceiled * channels};
+  const auto local = std::vector<size_t>{db_["COPY_DIMX"], db_["COPY_DIMY"]};
   RunKernel(kernel, queue_, device_, global, local, event_);
 }