From 6332ea99abd7ca7782cb22c8fafd938918536ca7 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Thu, 22 Feb 2018 21:13:08 +0000
Subject: [PATCH] one more fix

---
 src/convolutional_layer.c |    2 ++
 1 files changed, 2 insertions(+), 0 deletions(-)

diff --git a/src/convolutional_layer.c b/src/convolutional_layer.c
index 6865808..aeb621a 100644
--- a/src/convolutional_layer.c
+++ b/src/convolutional_layer.c
@@ -148,7 +148,9 @@
 	cudnnDataType_t data_type = CUDNN_DATA_FLOAT;
 #endif
 	// Tensor Core uses CUDNN_TENSOR_OP_MATH instead of CUDNN_DEFAULT_MATH
+#if(CUDNN_MAJOR >= 7)
 	cudnnSetConvolutionMathType(l->convDesc, CUDNN_TENSOR_OP_MATH);
+#endif
 
 	// INT8_CONFIG, INT8_EXT_CONFIG, INT8x4_CONFIG and INT8x4_EXT_CONFIG are only supported 
 	// on architectures with DP4A support (compute capability 6.1 and later).

--
Gitblit v1.10.0