diff --git a/mindspore/ccsrc/utils/context/context_extends.cc b/mindspore/ccsrc/utils/context/context_extends.cc
index b460a66c4d..172a7d1f05 100644
--- a/mindspore/ccsrc/utils/context/context_extends.cc
+++ b/mindspore/ccsrc/utils/context/context_extends.cc
@@ -24,6 +24,10 @@
 
 namespace mindspore {
 namespace context {
+#ifdef ENBALE_GE
+using mindspore::transform::DfGraphManager;
+#endif
+
 #ifndef NO_DLIB
 // Open tdt dataset
 bool OpenTsd(const std::shared_ptr<MsContext> &ms_context_ptr) {