make MKLDNN macro simple for imperative_utils.h (#16652)

apache · Oct 28, 2019 · a26ef14 · a26ef14
1 parent a9941ff
commit a26ef14
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/imperative/imperative_utils.h b/src/imperative/imperative_utils.h
@@ -484,10 +484,10 @@ inline void PushFComputeEx(const FComputeEx& fn,
         CreateDefaultInputs(inputs, &inputs_fallback);
         fn(attrs, opctx, inputs_fallback, req, outputs);
       } else {
-#endif
         fn(attrs, opctx, inputs, req, outputs);
-#if MXNET_USE_MKLDNN == 1
       }
+#else
+      fn(attrs, opctx, inputs, req, outputs);
 #endif
       if (ctx.dev_mask() == gpu::kDevMask && exec_type == ExecType::kSync && !rctx.is_bulk) {
         rctx.get_stream<gpu>()->Wait();
@@ -548,10 +548,10 @@ inline void PushOperator(const OpStatePtr& state,
         CreateDefaultInputs(inputs, &inputs_fallback);
         fcompute_ex(state, opctx, inputs_fallback, req, outputs);
       } else {
-#endif
         fcompute_ex(state, opctx, inputs, req, outputs);
-#if MXNET_USE_MKLDNN == 1
       }
+#else
+      fcompute_ex(state, opctx, inputs, req, outputs);
 #endif
       if (ctx.dev_mask() == gpu::kDevMask && exec_type == ExecType::kSync
           && rctx.get_stream<gpu>() && !rctx.is_bulk) {