From 880cf187d87c904f5fe574802ecff99118643f2d Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Fri, 09 Mar 2018 16:44:46 +0000
Subject: [PATCH] Fixed multi-GPU training for Tensor Cores
---
src/convolutional_layer.c | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/convolutional_layer.c b/src/convolutional_layer.c
index 377b898..7c0c00b 100644
--- a/src/convolutional_layer.c
+++ b/src/convolutional_layer.c
@@ -305,8 +305,8 @@
l.weights_gpu = cuda_make_array(l.weights, c*n*size*size);
#ifdef CUDNN_HALF
- l.weights_gpu16 = cuda_make_array(l.weights, c*n*size*size / 2);
- l.weight_updates_gpu16 = cuda_make_array(l.weight_updates, c*n*size*size / 2);
+ l.weights_gpu16 = cuda_make_array(NULL, c*n*size*size / 2); //cuda_make_array(l.weights, c*n*size*size / 2);
+ l.weight_updates_gpu16 = cuda_make_array(NULL, c*n*size*size / 2); //cuda_make_array(l.weight_updates, c*n*size*size / 2);
#endif
l.weight_updates_gpu = cuda_make_array(l.weight_updates, c*n*size*size);
--
Gitblit v1.10.0