change

Author: KaiXinXiaoLei

core/src/main/scala/org/apache/spark/scheduler/cluster/YarnSchedulerBackend.scala

@@ -91,7 +91,6 @@ private[spark] abstract class YarnSchedulerBackend(
       filterParams.foreach { case (k, v) => conf.set(s"spark.$filterName.param.$k", v) }
       scheduler.sc.ui.foreach { ui => JettyUtils.addFilters(ui.getHandlers, conf) }
     }
-    scheduler.sc.executorAllocationManager.foreach(_.reSetNumExecutorsTarget())
   }
 
   override def createDriverEndpoint(properties: Seq[(String, String)]): DriverEndpoint = {
@@ -171,6 +170,7 @@ private[spark] abstract class YarnSchedulerBackend(
       case RegisterClusterManager(am) =>
         logInfo(s"ApplicationMaster registered as $am")
         amEndpoint = Some(am)
+        scheduler.sc.executorAllocationManager.foreach(_.reSetNumExecutorsTarget())
 
       case AddWebUIFilter(filterName, filterParams, proxyBase) =>
         addWebUIFilter(filterName, filterParams, proxyBase)