Merge branch 'v1.x' into add-lts-nd-tests

apache · Jul 23, 2020 · b56fad4 · b56fad4
2 parents 8bbc53b + 91d535a
commit b56fad4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/c_api/c_api.cc b/src/c_api/c_api.cc
@@ -1315,6 +1315,7 @@ int MXNotifyShutdown() {
   API_BEGIN();
   mxnet::op::custom::CustomOperator::Get()->Stop();
   Engine::Get()->NotifyShutdown();
+  Engine::Get()->WaitForAll();
   API_END();
 }