log if flag is off

apache · Aug 15, 2018 · a3c94fd · a3c94fd
1 parent 06b2a5c
commit a3c94fd
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/executor/attach_op_execs_pass.cc b/src/executor/attach_op_execs_pass.cc
@@ -262,6 +262,11 @@ void CreateOpExecs(const Graph& g, OpExecVector* p_ret, size_t i) {
   const auto& vctx = g.GetAttr<ContextVector>("context");
   const auto& dispatch_modes = g.GetAttr<DispatchModeVector>("dispatch_mode");
 
+#if MXNET_USE_MKLDNN == 1
+  if (!MKLDNNEnvSet()) common::LogOnce("MXNET_MKLDNN_ENABLED flag is off");
+#endif
+
+
   // get the graph
   const auto& idx = g.indexed_graph();
   OpExecVector& ret = *p_ret;