From e0138d7980a532c5b7722e6337774561f2cc86ee Mon Sep 17 00:00:00 2001
From: wjm <wangjiming5@huawei.com>
Date: Sat, 27 Feb 2021 07:45:12 +0800
Subject: [PATCH] fix ut

---
 tests/ut/ge/graph/load/model_helper_unittest.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/ut/ge/graph/load/model_helper_unittest.cc b/tests/ut/ge/graph/load/model_helper_unittest.cc
index 5eacd9eb..03605dc7 100644
--- 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