From cad4d1618fee74471d335314cb77070fee951a42 Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Sun, 25 Feb 2018 13:29:44 +0000 Subject: [PATCH] Added support for Tensor Cores CC >= 7.0 (V100). For FP16/32 (mixed precision) define CUDNN_HALF should be used. --- src/deconvolutional_layer.h | 1 - 1 files changed, 0 insertions(+), 1 deletions(-) diff --git a/src/deconvolutional_layer.h b/src/deconvolutional_layer.h index b6af397..2d36e02 100644 --- a/src/deconvolutional_layer.h +++ b/src/deconvolutional_layer.h @@ -2,7 +2,6 @@ #define DECONVOLUTIONAL_LAYER_H #include "cuda.h" -#include "params.h" #include "image.h" #include "activations.h" #include "layer.h" -- Gitblit v1.10.0