MLIR-disabled compilcation fix

f4954d57 · nmostafa · 0354192c · f4954d57
Commit f4954d57 authored Jul 25, 2019 by nmostafa
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

cpu_runtime_context.hpp src/ngraph/runtime/cpu/cpu_runtime_context.hpp +4 -1

No files found.
--- a/src/ngraph/runtime/cpu/cpu_runtime_context.hpp
+++ b/src/ngraph/runtime/cpu/cpu_runtime_context.hpp
@@ -25,9 +25,12 @@
 #include <tbb/flow_graph.h>
 #include <tbb/global_control.h>
 #include <tbb/task_scheduler_init.h>
-#include "contrib/mlir/compiler.hpp"
 #include "ngraph/op/experimental/compiled_kernel.hpp"

+#ifdef NGRAPH_MLIR_ENABLE
+#include "contrib/mlir/compiler.hpp"
+#endif
+
 namespace mkldnn
 {
    class primitive;