From ae43c2bc32fbb838bfebeeaf2c2b058ccab5c83c Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@burninator.cs.washington.edu>
Date: Thu, 23 Jun 2016 05:31:14 +0000
Subject: [PATCH] hi
---
src/cuda.c | 26 ++++++++++++++++++++------
1 files changed, 20 insertions(+), 6 deletions(-)
diff --git a/src/cuda.c b/src/cuda.c
index e95feff..327813d 100644
--- a/src/cuda.c
+++ b/src/cuda.c
@@ -46,6 +46,19 @@
return d;
}
+#ifdef CUDNN
+cudnnHandle_t cudnn_handle()
+{
+ static int init = 0;
+ static cudnnHandle_t handle;
+ if(!init) {
+ cudnnCreate(&handle);
+ init = 1;
+ }
+ return handle;
+}
+#endif
+
cublasHandle_t blas_handle()
{
static int init = 0;
@@ -57,7 +70,7 @@
return handle;
}
-float *cuda_make_array(float *x, int n)
+float *cuda_make_array(float *x, size_t n)
{
float *x_gpu;
size_t size = sizeof(float)*n;
@@ -67,10 +80,11 @@
status = cudaMemcpy(x_gpu, x, size, cudaMemcpyHostToDevice);
check_error(status);
}
+ if(!x_gpu) error("Cuda malloc failed\n");
return x_gpu;
}
-void cuda_random(float *x_gpu, int n)
+void cuda_random(float *x_gpu, size_t n)
{
static curandGenerator_t gen;
static int init = 0;
@@ -83,7 +97,7 @@
check_error(cudaPeekAtLastError());
}
-float cuda_compare(float *x_gpu, float *x, int n, char *s)
+float cuda_compare(float *x_gpu, float *x, size_t n, char *s)
{
float *tmp = calloc(n, sizeof(float));
cuda_pull_array(x_gpu, tmp, n);
@@ -96,7 +110,7 @@
return err;
}
-int *cuda_make_int_array(int n)
+int *cuda_make_int_array(size_t n)
{
int *x_gpu;
size_t size = sizeof(int)*n;
@@ -111,14 +125,14 @@
check_error(status);
}
-void cuda_push_array(float *x_gpu, float *x, int n)
+void cuda_push_array(float *x_gpu, float *x, size_t n)
{
size_t size = sizeof(float)*n;
cudaError_t status = cudaMemcpy(x_gpu, x, size, cudaMemcpyHostToDevice);
check_error(status);
}
-void cuda_pull_array(float *x_gpu, float *x, int n)
+void cuda_pull_array(float *x_gpu, float *x, size_t n)
{
size_t size = sizeof(float)*n;
cudaError_t status = cudaMemcpy(x, x_gpu, size, cudaMemcpyDeviceToHost);
--
Gitblit v1.10.0