From 01c742f0025d195a1f265981b52551fa14f1d484 Mon Sep 17 00:00:00 2001
From: Edmond Yoo <hj3yoo@uwaterloo.ca>
Date: Thu, 06 Sep 2018 16:51:53 +0000
Subject: [PATCH] build, custom configs, training results so far
---
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