Compile fix

src/connected_layer.c

@@ -147,7 +147,7 @@ connected_layer make_connected_layer(int batch, int steps, int inputs, int outpu
     }
 #ifdef CUDNN
     create_convolutional_cudnn_tensors(&l);
-    cudnn_convolutional_setup(&l, cudnn_fastest);   // cudnn_fastest, cudnn_smallest
+    cudnn_convolutional_setup(&l, cudnn_fastest, 0);   // cudnn_fastest, cudnn_smallest
     l.workspace_size = get_connected_workspace_size(l);
 #endif  // CUDNN
 #endif  // GPU

src/convolutional_layer.h

@@ -23,7 +23,7 @@ void pull_convolutional_layer(convolutional_layer layer);
 void add_bias_gpu(float *output, float *biases, int batch, int n, int size);
 void backward_bias_gpu(float *bias_updates, float *delta, int batch, int n, int size);
 #ifdef CUDNN
-void cudnn_convolutional_setup(layer *l, int cudnn_preference);
+void cudnn_convolutional_setup(layer *l, int cudnn_preference, size_t workspace_size_specify);
 void create_convolutional_cudnn_tensors(layer *l);
 void cuda_convert_f32_to_f16(float* input_f32, size_t size, float *output_f16);
 #endif

src/network.c

@@ -472,7 +472,7 @@ void set_batch_network(network *net, int b)
 
 #ifdef CUDNN
         if(net->layers[i].type == CONVOLUTIONAL){
-            cudnn_convolutional_setup(net->layers + i, cudnn_fastest);
+            cudnn_convolutional_setup(net->layers + i, cudnn_fastest, 0);
         }
         else if (net->layers[i].type == MAXPOOL) {
             cudnn_maxpool_setup(net->layers + i);