Fixed gpu_id for DLL/SO

AlexeyAB

2018-03-19 ced198e9390195875d743d77eadece99c7fd5b38

 src/layer.h

@@ -34,6 +34,7 @@
    XNOR,
    REGION,
    REORG,
   REORG_OLD,
    BLANK
} LAYER_TYPE;

@@ -41,6 +42,18 @@
    SSE, MASKED, SMOOTH
} COST_TYPE;

typedef struct {
   int batch;
   float learning_rate;
   float momentum;
   float decay;
   int adam;
   float B1;
   float B2;
   float eps;
   int t;
} update_args;

struct layer{
    LAYER_TYPE type;
    ACTIVATION activation;
@@ -84,6 +97,7 @@
    float exposure;
    float shift;
    float ratio;
   int focal_loss;
    int softmax;
    int classes;
    int coords;
@@ -242,6 +256,9 @@
    float * weights_gpu;
    float * weight_updates_gpu;

   float * weights_gpu16;
   float * weight_updates_gpu16;

    float * biases_gpu;
    float * bias_updates_gpu;

@@ -256,6 +273,7 @@
    #ifdef CUDNN
    cudnnTensorDescriptor_t srcTensorDesc, dstTensorDesc;
    cudnnTensorDescriptor_t dsrcTensorDesc, ddstTensorDesc;
   cudnnTensorDescriptor_t normTensorDesc;
    cudnnFilterDescriptor_t weightDesc;
    cudnnFilterDescriptor_t dweightDesc;
    cudnnConvolutionDescriptor_t convDesc;