From 0f645836f193e75c4c3b718369e6fab15b5d19c5 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 11 Feb 2015 03:41:03 +0000
Subject: [PATCH] Detection is back, baby\!
---
src/col2im_kernels.cu | 8 ++++----
1 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/src/col2im_kernels.cu b/src/col2im_kernels.cu
index 73de9b7..2fa2030 100644
--- a/src/col2im_kernels.cu
+++ b/src/col2im_kernels.cu
@@ -3,7 +3,7 @@
#include "cuda.h"
}
-__global__ void col2im_kernel(float *data_col, int offset,
+__global__ void col2im_kernel(float *data_col,
int channels, int height, int width,
int ksize, int stride, int pad, float *data_im)
{
@@ -46,17 +46,17 @@
val += part;
}
}
- data_im[index+offset] = val;
+ data_im[index] = val;
}
-extern "C" void col2im_ongpu(float *data_col, int offset,
+extern "C" void col2im_ongpu(float *data_col,
int channels, int height, int width,
int ksize, int stride, int pad, float *data_im)
{
size_t n = channels*height*width;
- col2im_kernel<<<cuda_gridsize(n), BLOCK>>>(data_col, offset, channels, height, width, ksize, stride, pad, data_im);
+ col2im_kernel<<<cuda_gridsize(n), BLOCK>>>(data_col, channels, height, width, ksize, stride, pad, data_im);
check_error(cudaPeekAtLastError());
}
--
Gitblit v1.10.0