From 74f8ede97638757268b84cabe4276ecab049798e Mon Sep 17 00:00:00 2001 From: hedaoyuan <hedaoyuan@github.com> Date: Tue, 1 Aug 2017 20:54:43 +0800 Subject: [PATCH] Fix when use_nnpack and isDeconv == true. --- paddle/gserver/layers/ExpandConvLayer.cpp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/paddle/gserver/layers/ExpandConvLayer.cpp b/paddle/gserver/layers/ExpandConvLayer.cpp index 783e02e47c..0ece279931 100644 --- a/paddle/gserver/layers/ExpandConvLayer.cpp +++ b/paddle/gserver/layers/ExpandConvLayer.cpp @@ -57,8 +57,7 @@ bool ExpandConvLayer::init(const LayerMap &layerMap, convGradFilterType = "GemmConvGradFilter"; } - if (FLAGS_use_nnpack) { - CHECK_EQ(isDeconv_, false); + if (FLAGS_use_nnpack && !isDeconv_) { createFunction(forward_, "NNPACKConv", FuncConfig() -- GitLab