From 3459c47bfa02d23d300df9a44ba3ec9c958850fe Mon Sep 17 00:00:00 2001
From: Philip Kahn <tigerhawkvok@gmail.com>
Date: Fri, 04 May 2018 21:00:11 +0000
Subject: [PATCH] Also replace root version to Python3
---
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