From 483fdfa951feeb20903da1598cb9d611b93de6ca Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Sun, 03 Jun 2018 11:21:15 +0000
Subject: [PATCH] Fix for multi-GPU and DGX

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

diff --git a/src/layer.c b/src/layer.c
index 582cbb3..3b47917 100644
--- a/src/layer.c
+++ b/src/layer.c
@@ -11,6 +11,7 @@
 #endif
 		return;
 	}
+	if (l.mask)               free(l.mask);
 	if (l.cweights)           free(l.cweights);
 	if (l.indexes)            free(l.indexes);
 	if (l.input_layers)       free(l.input_layers);

--
Gitblit v1.10.0