Go to the documentation of this file.00001 #ifndef CUDA_H
00002 #define CUDA_H
00003
00004 extern int gpu_index;
00005
00006 #ifdef GPU
00007
00008 #define BLOCK 512
00009
00010 #include "cuda_runtime.h"
00011 #include "curand.h"
00012 #include "cublas_v2.h"
00013
00014 #ifdef CUDNN
00015 #include "cudnn.h"
00016 #endif
00017
00018 void check_error(cudaError_t status);
00019 cublasHandle_t blas_handle();
00020 float *cuda_make_array(float *x, size_t n);
00021 int *cuda_make_int_array(size_t n);
00022 void cuda_push_array(float *x_gpu, float *x, size_t n);
00023 void cuda_pull_array(float *x_gpu, float *x, size_t n);
00024 void cuda_set_device(int n);
00025 void cuda_free(float *x_gpu);
00026 void cuda_random(float *x_gpu, size_t n);
00027 float cuda_compare(float *x_gpu, float *x, size_t n, char *s);
00028 dim3 cuda_gridsize(size_t n);
00029
00030 #ifdef CUDNN
00031 cudnnHandle_t cudnn_handle();
00032 #endif
00033
00034 #endif
00035 #endif