Fix when use_nnpack and isDeconv == true.

8 years ago · 74f8ede976
parent 953f8ddf05
commit 74f8ede976
1 changed files with 1 additions and 2 deletions
--- 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()