summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/kernels/levelx/im2col.opencl4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/kernels/levelx/im2col.opencl b/src/kernels/levelx/im2col.opencl
index a64d6538..e7f69420 100644
--- a/src/kernels/levelx/im2col.opencl
+++ b/src/kernels/levelx/im2col.opencl
@@ -37,8 +37,8 @@ void im2col(const int input_h, const int input_w, const int channels,
// Thread IDs
const int w_id = get_global_id(0); // image width, max 'output_w'
- const int h_id = get_global_id(1) % output_h; // image height, max 'output_h'
- const int c_id = get_global_id(1) / output_h; // input channels
+ const int h_id = ((int)get_global_id(1)) % output_h; // image height, max 'output_h'
+ const int c_id = ((int)get_global_id(1)) / output_h; // input channels
if (h_id < output_h && w_id < output_w && c_id < channels) {
#pragma unroll