Browse Source

add ENABLE_GE

tags/v0.6.0-beta
jinyaohui 5 years ago
parent
commit
c08759504c
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/ccsrc/pipeline/pipeline.cc

+ 2
- 0
mindspore/ccsrc/pipeline/pipeline.cc View File

@@ -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_;


Loading…
Cancel
Save