From 2cd95dc51750f69527af1da5115abfce0254101a Mon Sep 17 00:00:00 2001 From: raver119 Date: Wed, 24 Jul 2019 14:45:09 +0300 Subject: [PATCH] upsampling2d fix CUDA Signed-off-by: raver119 --- libnd4j/include/ops/declarable/helpers/cuda/convolutions.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libnd4j/include/ops/declarable/helpers/cuda/convolutions.cu b/libnd4j/include/ops/declarable/helpers/cuda/convolutions.cu index 1fef8696a..e993b370e 100644 --- a/libnd4j/include/ops/declarable/helpers/cuda/convolutions.cu +++ b/libnd4j/include/ops/declarable/helpers/cuda/convolutions.cu @@ -1362,7 +1362,7 @@ static void depthwiseConv2dBP_(const NDArray* input, const NDArray* weights, con gradBR = new NDArray(gradB->reshape(gradB->ordering(), {(int)gradB->lengthOf()})); gradO->reduceAlongDimension(reduce::Sum, gradBR, {0,indOoH,indOoH+1}); // sum over bS, oH, oW if(gradBR != gradB) - delete gradB; + delete gradBR; } //----- calculation of gradI -----//