diff --git a/mindspore/ccsrc/utils/context/context_extends.cc b/mindspore/ccsrc/utils/context/context_extends.cc
index b460a66c4de2923955da3be10875f34910d67924..172a7d1f05cc2b86cfb1033d435e3e296a9004fa 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) {