From 6194d43efba30aac90a64676e7770f020e4a5588 Mon Sep 17 00:00:00 2001 From: Cedric Nugteren Date: Thu, 31 Aug 2017 20:34:10 +0200 Subject: Fixed a bug in im2col confusing first and second workgroup size; made im2col kernel 2d instead of 3d --- src/routines/levelx/xim2col.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'src/routines') 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::DoIm2col(const size_t channels, const size_t height, const size kernel.SetArgument(15, static_cast(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{w_ceiled, h_ceiled, channels}; - const auto local = std::vector{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{w_ceiled, h_ceiled * channels}; + const auto local = std::vector{db_["COPY_DIMX"], db_["COPY_DIMY"]}; RunKernel(kernel, queue_, device_, global, local, event_); } -- cgit v1.2.3