activation_debug

luobao-intel · Jul 30, 2018 · 4cd4ae8 · 4cd4ae8
1 parent 18e73c1
commit 4cd4ae8
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/operator/nn/activation.cc b/src/operator/nn/activation.cc
@@ -139,8 +139,7 @@ inline static bool BackwardActStorageType(const nnvm::NodeAttrs& attrs,
 #if MXNET_USE_CUDNN == 1
   ret = ElemwiseStorageAttr<false, false, false>(attrs, dev_mask, dispatch_mode,
                                                  in_attrs, out_attrs);
-#endif
-#if MXNET_USE_MKLDNN == 1
+#elif MXNET_USE_MKLDNN == 1
   bool should_enter = SupportMKLDNNAct(param);
   ret = MKLDNNStorageType(attrs, dev_mask, should_enter, dispatch_mode,
                           in_attrs, out_attrs);