From 2313a8eb54d703323279c0fb9b2c9c52d26f0cf9 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Fri, 06 Mar 2015 18:49:03 +0000 Subject: [PATCH] Split commands into different files --- src/cuda.h | 7 +++++-- 1 files changed, 5 insertions(+), 2 deletions(-) diff --git a/src/cuda.h b/src/cuda.h index 08c0340..cbe7975 100644 --- a/src/cuda.h +++ b/src/cuda.h @@ -1,13 +1,15 @@ #ifndef CUDA_H #define CUDA_H +extern int gpu_index; + +#ifdef GPU + #define BLOCK 256 #include "cuda_runtime.h" #include "cublas_v2.h" -extern int gpu_index; - void check_error(cudaError_t status); cublasHandle_t blas_handle(); float *cuda_make_array(float *x, int n); @@ -19,3 +21,4 @@ dim3 cuda_gridsize(size_t n); #endif +#endif -- Gitblit v1.10.0