fix class not declear when build ge

5 years ago · 344bcd85be
parent 7113f1f22c
commit 344bcd85be
1 changed files with 4 additions and 0 deletions
--- 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) {