From 0f7f2899b65343e56b0a1188f703d459d824d398 Mon Sep 17 00:00:00 2001 From: Joseph Redmon Date: Sun, 15 Nov 2015 19:51:26 -0800 Subject: [PATCH] Fix for cuda 7.5 --- src/activation_kernels.cu | 4 ++++ src/avgpool_layer_kernels.cu | 4 ++++ src/blas_kernels.cu | 4 ++++ src/coco_kernels.cu | 4 ++++ src/col2im_kernels.cu | 4 ++++ src/convolutional_kernels.cu | 4 ++++ src/crop_layer_kernels.cu | 4 ++++ src/deconvolutional_kernels.cu | 4 ++++ src/dropout_layer_kernels.cu | 4 ++++ src/im2col_kernels.cu | 4 ++++ src/local_kernels.cu | 4 ++++ src/maxpool_layer_kernels.cu | 4 ++++ src/network_kernels.cu | 4 ++++ src/softmax_layer_kernels.cu | 4 ++++ src/yolo_kernels.cu | 4 ++++ 15 files changed, 60 insertions(+) diff --git a/src/activation_kernels.cu b/src/activation_kernels.cu index 16dd4d2a..0ab9fd96 100644 --- a/src/activation_kernels.cu +++ b/src/activation_kernels.cu @@ -1,3 +1,7 @@ +#include "cuda_runtime.h" +#include "curand.h" +#include "cublas_v2.h" + extern "C" { #include "activations.h" #include "cuda.h" diff --git a/src/avgpool_layer_kernels.cu b/src/avgpool_layer_kernels.cu index 1bd2a2af..b7e2770e 100644 --- a/src/avgpool_layer_kernels.cu +++ b/src/avgpool_layer_kernels.cu @@ -1,3 +1,7 @@ +#include "cuda_runtime.h" +#include "curand.h" +#include "cublas_v2.h" + extern "C" { #include "avgpool_layer.h" #include "cuda.h" diff --git a/src/blas_kernels.cu b/src/blas_kernels.cu index b990ca33..4da31d14 100644 --- a/src/blas_kernels.cu +++ b/src/blas_kernels.cu @@ -1,3 +1,7 @@ +#include "cuda_runtime.h" +#include "curand.h" +#include "cublas_v2.h" + extern "C" { #include "blas.h" #include "cuda.h" diff --git a/src/coco_kernels.cu b/src/coco_kernels.cu index 9c201c0a..a3b4875c 100644 --- a/src/coco_kernels.cu +++ b/src/coco_kernels.cu @@ -1,3 +1,7 @@ +#include "cuda_runtime.h" +#include "curand.h" +#include "cublas_v2.h" + extern "C" { #include "network.h" #include "detection_layer.h" diff --git a/src/col2im_kernels.cu b/src/col2im_kernels.cu index 7262f923..b381f1a7 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" diff --git a/src/convolutional_kernels.cu b/src/convolutional_kernels.cu index 60a18795..5f24ca5e 100644 --- a/src/convolutional_kernels.cu +++ b/src/convolutional_kernels.cu @@ -1,3 +1,7 @@ +#include "cuda_runtime.h" +#include "curand.h" +#include "cublas_v2.h" + extern "C" { #include "convolutional_layer.h" #include "gemm.h" diff --git a/src/crop_layer_kernels.cu b/src/crop_layer_kernels.cu index fdebd1b3..8891030a 100644 --- a/src/crop_layer_kernels.cu +++ b/src/crop_layer_kernels.cu @@ -1,3 +1,7 @@ +#include "cuda_runtime.h" +#include "curand.h" +#include "cublas_v2.h" + extern "C" { #include "crop_layer.h" #include "utils.h" diff --git a/src/deconvolutional_kernels.cu b/src/deconvolutional_kernels.cu index a74fb782..68cc839e 100644 --- a/src/deconvolutional_kernels.cu +++ b/src/deconvolutional_kernels.cu @@ -1,3 +1,7 @@ +#include "cuda_runtime.h" +#include "curand.h" +#include "cublas_v2.h" + extern "C" { #include "convolutional_layer.h" #include "deconvolutional_layer.h" diff --git a/src/dropout_layer_kernels.cu b/src/dropout_layer_kernels.cu index 3dfa6410..f65e3c3d 100644 --- a/src/dropout_layer_kernels.cu +++ b/src/dropout_layer_kernels.cu @@ -1,3 +1,7 @@ +#include "cuda_runtime.h" +#include "curand.h" +#include "cublas_v2.h" + extern "C" { #include "dropout_layer.h" #include "cuda.h" diff --git a/src/im2col_kernels.cu b/src/im2col_kernels.cu index d122748a..c2dd7802 100644 --- a/src/im2col_kernels.cu +++ b/src/im2col_kernels.cu @@ -1,3 +1,7 @@ +#include "cuda_runtime.h" +#include "curand.h" +#include "cublas_v2.h" + extern "C" { #include "im2col.h" #include "cuda.h" diff --git a/src/local_kernels.cu b/src/local_kernels.cu index 87174168..14e5a0ed 100644 --- a/src/local_kernels.cu +++ b/src/local_kernels.cu @@ -1,3 +1,7 @@ +#include "cuda_runtime.h" +#include "curand.h" +#include "cublas_v2.h" + extern "C" { #include "local_layer.h" #include "gemm.h" diff --git a/src/maxpool_layer_kernels.cu b/src/maxpool_layer_kernels.cu index 8f69f905..331a5e2c 100644 --- a/src/maxpool_layer_kernels.cu +++ b/src/maxpool_layer_kernels.cu @@ -1,3 +1,7 @@ +#include "cuda_runtime.h" +#include "curand.h" +#include "cublas_v2.h" + extern "C" { #include "maxpool_layer.h" #include "cuda.h" diff --git a/src/network_kernels.cu b/src/network_kernels.cu index ffd5c590..26b84041 100644 --- a/src/network_kernels.cu +++ b/src/network_kernels.cu @@ -1,3 +1,7 @@ +#include "cuda_runtime.h" +#include "curand.h" +#include "cublas_v2.h" + extern "C" { #include #include diff --git a/src/softmax_layer_kernels.cu b/src/softmax_layer_kernels.cu index 7e133870..66371805 100644 --- a/src/softmax_layer_kernels.cu +++ b/src/softmax_layer_kernels.cu @@ -1,3 +1,7 @@ +#include "cuda_runtime.h" +#include "curand.h" +#include "cublas_v2.h" + extern "C" { #include "softmax_layer.h" #include "cuda.h" diff --git a/src/yolo_kernels.cu b/src/yolo_kernels.cu index 487e9bd9..a0ff5f76 100644 --- a/src/yolo_kernels.cu +++ b/src/yolo_kernels.cu @@ -1,3 +1,7 @@ +#include "cuda_runtime.h" +#include "curand.h" +#include "cublas_v2.h" + extern "C" { #include "network.h" #include "detection_layer.h"