only stop nm when sucess

2015-03-15 23:02:15 -07:00 · 2015-03-15 23:02:15 -07:00 · b2505e3d6f
commit b2505e3d6f
parent bc696c9273
1 changed files with 3 additions and 1 deletions
--- a/yarn/src/org/apache/hadoop/yarn/rabit/ApplicationMaster.java
+++ b/yarn/src/org/apache/hadoop/yarn/rabit/ApplicationMaster.java
@ -216,7 +216,9 @@ public class ApplicationMaster {
            assert (killedTasks.size() + finishedTasks.size() == numTasks);
            success = finishedTasks.size() == numTasks;
            LOG.info("Application completed. Stopping running containers");
-            nmClient.stop();
+            if (success) {
+              nmClient.stop();
+            }
            diagnostics = "Diagnostics." + ", num_tasks" + this.numTasks
                    + ", finished=" + this.finishedTasks.size() + ", failed="
                    + this.killedTasks.size() + "\n" + this.abortDiagnosis;