Added support for Tensor Cores CC >= 7.0 (V100). For FP16/32 (mixed precisi...

AlexeyAB

2018-02-25 cad4d1618fee74471d335314cb77070fee951a42

 src/network.c

@@ -316,6 +316,8 @@
        net->layers[i].batch = b;
#ifdef CUDNN
        if(net->layers[i].type == CONVOLUTIONAL){
         cudnn_convolutional_setup(net->layers + i, cudnn_fastest);
         /*
         layer *l = net->layers + i;
            cudnn_convolutional_setup(l, cudnn_fastest);
         // check for excessive memory consumption 
@@ -327,6 +329,7 @@
            cudnn_convolutional_setup(l, cudnn_smallest);
            l->workspace_size = get_workspace_size(*l);
         }
         */
        }
#endif
    }