From a2b1785ac6df619ae7b6e47e31b988853729baa3 Mon Sep 17 00:00:00 2001 From: Sergio Date: Sun, 13 Apr 2014 20:11:49 -0700 Subject: [PATCH] Set bottom_diff to 0 and remove Async memcopy --- src/caffe/layers/pooling_layer.cu | 1 + 1 file changed, 1 insertion(+) diff --git a/src/caffe/layers/pooling_layer.cu b/src/caffe/layers/pooling_layer.cu index 95a754f..77126e6 100644 --- a/src/caffe/layers/pooling_layer.cu +++ b/src/caffe/layers/pooling_layer.cu @@ -327,6 +327,7 @@ void PoolingLayer::Backward_gpu(const vector*>& top, // Since we have the mask we only need count top_diff count = top[0]->count(); // NOLINT_NEXT_LINE(whitespace/operators) + caffe_gpu_memset(count,Dtype(0.),bottom_diff); MaxPoolBackward<<>>( count, top_diff, top[0]->num(), channels_, height_, width_, pooled_height_, pooled_width_, -- 2.7.4