From be9d971ddb9ea0520da78cfff7788eb5481f095e Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Tue, 3 Apr 2018 16:56:53 +0300 Subject: [PATCH] Compile error fix --- src/network.c | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/network.c b/src/network.c index bfade3cc..d532b859 100644 --- a/src/network.c +++ b/src/network.c @@ -775,7 +775,11 @@ void fuse_conv_batchnorm(network net) } l->batch_normalize = 0; - push_convolutional_layer(*l); +#ifdef GPU + if (gpu_index >= 0) { + push_convolutional_layer(*l); + } +#endif } } else {