|
@@ -51,6 +51,7 @@ import org.apache.hadoop.yarn.api.records.Resource;
|
|
|
import org.apache.hadoop.yarn.conf.YarnConfiguration;
|
|
|
import org.apache.hadoop.yarn.exceptions.YarnException;
|
|
|
import org.apache.hadoop.yarn.server.resourcemanager.ResourceManager;
|
|
|
+import org.apache.hadoop.yarn.server.resourcemanager.monitor.capacity.ProportionalCapacityPreemptionPolicy;
|
|
|
import org.apache.hadoop.yarn.server.resourcemanager.rmnode.RMNode;
|
|
|
import org.apache.hadoop.yarn.server.utils.BuilderUtils;
|
|
|
import org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.CapacityScheduler;
|
|
@@ -171,6 +172,9 @@ public class SLSRunner {
|
|
|
if(Class.forName(schedulerClass) == CapacityScheduler.class) {
|
|
|
rmConf.set(YarnConfiguration.RM_SCHEDULER,
|
|
|
SLSCapacityScheduler.class.getName());
|
|
|
+ rmConf.setBoolean(YarnConfiguration.RM_SCHEDULER_ENABLE_MONITORS, true);
|
|
|
+ rmConf.set(YarnConfiguration.RM_SCHEDULER_MONITOR_POLICIES,
|
|
|
+ ProportionalCapacityPreemptionPolicy.class.getName());
|
|
|
} else {
|
|
|
rmConf.set(YarnConfiguration.RM_SCHEDULER,
|
|
|
ResourceSchedulerWrapper.class.getName());
|