From 75d9b8f2846af2621ecd39c34dd09fc245ed1856 Mon Sep 17 00:00:00 2001 From: HeZheng Date: Thu, 1 Aug 2019 11:41:02 +0800 Subject: [PATCH] assert nothing if(input2.size(1) != input2.size(1)) return error;???? if(input2.size(1) != input3.size(1)) return error; right? --- my_package/SeparableConv/separableconv_cuda.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/my_package/SeparableConv/separableconv_cuda.cc b/my_package/SeparableConv/separableconv_cuda.cc index 1fb50a8..ce12438 100644 --- a/my_package/SeparableConv/separableconv_cuda.cc +++ b/my_package/SeparableConv/separableconv_cuda.cc @@ -21,7 +21,7 @@ int SeparableConvLayer_gpu_forward( if(channel!=3) return error; int batch = input1.size(0); if(input2.size( 0) != batch) return error; - if(input2.size(1) != input2.size(1)) return error; + if(input2.size(1) != input3.size(1)) return error; //change by zhenghe, am I right? int h = input1.size(2); int w = input1.size(3);