fix ge compile error

4 years ago · f9ab9106de
parent f34c8a45c5
commit f9ab9106de
1 changed files with 3 additions and 0 deletions
--- 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";