add ENABLE_GE

5 years ago · 0332e4a854
parent 927278be44
commit 0332e4a854
1 changed files with 2 additions and 0 deletions
--- a/mindspore/ccsrc/pipeline/pipeline.cc
+++ b/mindspore/ccsrc/pipeline/pipeline.cc
@ -245,7 +245,9 @@ py::dict ExecutorPy::GetAllreduceFusion(const std::string &phase) {
 }

 void ExecutorPy::DelNetRes(const std::string &id) {
+#ifdef ENABLE_GE
  FinalizeBackend();
+#endif
  if (executor_ != nullptr) {
    bool flag = false;
    auto tmp_info = info_;