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"