diff --git a/tests/ut/ge/hybrid/node_executor/aicore/aicore_task_compiler_unittest.cc b/tests/ut/ge/hybrid/node_executor/aicore/aicore_task_compiler_unittest.cc
index 7ab6bd8d..52a1339a 100644
--- a/tests/ut/ge/hybrid/node_executor/aicore/aicore_task_compiler_unittest.cc
+++ b/tests/ut/ge/hybrid/node_executor/aicore/aicore_task_compiler_unittest.cc
@@ -22,6 +22,8 @@
 #define protected public
 #include "init/gelib.h"
 #include "hybrid/node_executor/aicore/aicore_task_compiler.h"
+#undef private
+#undef protected
 
 using namespace std;
 using namespace testing;