From 9207607a595fdf6c2e2cb01473b6e08f0746b8d3 Mon Sep 17 00:00:00 2001
From: iovodov <b@ovdv.ru>
Date: Sat, 21 Apr 2018 20:22:16 +0000
Subject: [PATCH] Fix issue #674: parameter "flip" is added to [net] parameters for classifier and detector
---
src/col2im_kernels.cu | 5 ++++-
1 files changed, 4 insertions(+), 1 deletions(-)
diff --git a/src/col2im_kernels.cu b/src/col2im_kernels.cu
index 7262f92..aed2df9 100644
--- a/src/col2im_kernels.cu
+++ b/src/col2im_kernels.cu
@@ -1,3 +1,7 @@
+#include "cuda_runtime.h"
+#include "curand.h"
+#include "cublas_v2.h"
+
extern "C" {
#include "col2im.h"
#include "cuda.h"
@@ -42,7 +46,6 @@
int ksize, int stride, int pad, float *data_im){
// We are going to launch channels * height_col * width_col kernels, each
// kernel responsible for copying a single-channel grid.
- pad = pad ? ksize/2 : 0;
int height_col = (height + 2 * pad - ksize) / stride + 1;
int width_col = (width + 2 * pad - ksize) / stride + 1;
int num_kernels = channels * height * width;
--
Gitblit v1.10.0