fix ut

4 years ago · e0138d7980
parent b83f971b7a
commit e0138d7980
1 changed files with 2 additions and 0 deletions
--- a/tests/ut/ge/graph/load/model_helper_unittest.cc
+++ b/tests/ut/ge/graph/load/model_helper_unittest.cc
@ -18,6 +18,8 @@
 #define private public
 #define protected public
 #include "framework/common/helper/model_helper.h"
+#include "framework/omg/model_tool.h"
+#include "framework/omg/ge_init.h"
 #include "ge/model/ge_model.h"
 #undef private
 #undef protected