Fix for cuda 7.5

pull/5299/head
Joseph Redmon 10 years ago
parent 0cd2379e2c
commit 0f7f2899b6
  1. 4
      src/activation_kernels.cu
  2. 4
      src/avgpool_layer_kernels.cu
  3. 4
      src/blas_kernels.cu
  4. 4
      src/coco_kernels.cu
  5. 4
      src/col2im_kernels.cu
  6. 4
      src/convolutional_kernels.cu
  7. 4
      src/crop_layer_kernels.cu
  8. 4
      src/deconvolutional_kernels.cu
  9. 4
      src/dropout_layer_kernels.cu
  10. 4
      src/im2col_kernels.cu
  11. 4
      src/local_kernels.cu
  12. 4
      src/maxpool_layer_kernels.cu
  13. 4
      src/network_kernels.cu
  14. 4
      src/softmax_layer_kernels.cu
  15. 4
      src/yolo_kernels.cu

@ -1,3 +1,7 @@
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
extern "C" { extern "C" {
#include "activations.h" #include "activations.h"
#include "cuda.h" #include "cuda.h"

@ -1,3 +1,7 @@
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
extern "C" { extern "C" {
#include "avgpool_layer.h" #include "avgpool_layer.h"
#include "cuda.h" #include "cuda.h"

@ -1,3 +1,7 @@
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
extern "C" { extern "C" {
#include "blas.h" #include "blas.h"
#include "cuda.h" #include "cuda.h"

@ -1,3 +1,7 @@
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
extern "C" { extern "C" {
#include "network.h" #include "network.h"
#include "detection_layer.h" #include "detection_layer.h"

@ -1,3 +1,7 @@
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
extern "C" { extern "C" {
#include "col2im.h" #include "col2im.h"
#include "cuda.h" #include "cuda.h"

@ -1,3 +1,7 @@
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
extern "C" { extern "C" {
#include "convolutional_layer.h" #include "convolutional_layer.h"
#include "gemm.h" #include "gemm.h"

@ -1,3 +1,7 @@
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
extern "C" { extern "C" {
#include "crop_layer.h" #include "crop_layer.h"
#include "utils.h" #include "utils.h"

@ -1,3 +1,7 @@
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
extern "C" { extern "C" {
#include "convolutional_layer.h" #include "convolutional_layer.h"
#include "deconvolutional_layer.h" #include "deconvolutional_layer.h"

@ -1,3 +1,7 @@
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
extern "C" { extern "C" {
#include "dropout_layer.h" #include "dropout_layer.h"
#include "cuda.h" #include "cuda.h"

@ -1,3 +1,7 @@
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
extern "C" { extern "C" {
#include "im2col.h" #include "im2col.h"
#include "cuda.h" #include "cuda.h"

@ -1,3 +1,7 @@
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
extern "C" { extern "C" {
#include "local_layer.h" #include "local_layer.h"
#include "gemm.h" #include "gemm.h"

@ -1,3 +1,7 @@
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
extern "C" { extern "C" {
#include "maxpool_layer.h" #include "maxpool_layer.h"
#include "cuda.h" #include "cuda.h"

@ -1,3 +1,7 @@
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
extern "C" { extern "C" {
#include <stdio.h> #include <stdio.h>
#include <time.h> #include <time.h>

@ -1,3 +1,7 @@
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
extern "C" { extern "C" {
#include "softmax_layer.h" #include "softmax_layer.h"
#include "cuda.h" #include "cuda.h"

@ -1,3 +1,7 @@
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
extern "C" { extern "C" {
#include "network.h" #include "network.h"
#include "detection_layer.h" #include "detection_layer.h"

Loading…
Cancel
Save