diff --git a/mindspore/ccsrc/vm/graph_partition.cc b/mindspore/ccsrc/vm/graph_partition.cc
index 5547e5a5e4..8384ee190f 100644
--- a/mindspore/ccsrc/vm/graph_partition.cc
+++ b/mindspore/ccsrc/vm/graph_partition.cc
@@ -25,6 +25,9 @@
 #include "base/core_ops.h"
 #include "utils/utils.h"
 #include "utils/ms_context.h"
+#ifdef ENABLE_GE
+#include "transform/graph_ir/convert.h"
+#endif
 namespace mindspore {
 const char kMsConvert[] = "ms";
 const char kMsVm[] = "vm";