diff options
| author | Cedric Nugteren <web@cedricnugteren.nl> | 2017-09-05 20:08:00 +0200 |
|---|---|---|
| committer | Cedric Nugteren <web@cedricnugteren.nl> | 2017-09-05 20:08:00 +0200 |
| commit | bb947890dec90712c92028c20234eafd48e6fa3e (patch) | |
| tree | 174b8568736373534e4902538710b5733b91ad81 /src | |
| parent | 28462aa05068522733dcdc812795ec235c40f7e1 (diff) | |
| parent | 8905da259df201919f86ca0c5be9dc1a577ac038 (diff) | |
Merge branch 'im2col_bugfix'
Diffstat (limited to 'src')
| -rw-r--r-- | src/kernels/levelx/im2col.opencl | 4 |
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 |