Merge -c 1442840 from trunk to branch-2 to fix YARN-370. Fix SchedulerUtils to correctly round up the resource for containers. Contributed by Zhijie Shen.
git-svn-id: https://svn.apache.org/repos/asf/hadoop/common/branches/branch-2@1442841 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
parent
f817236ff2
commit
807f694973
|
@ -198,6 +198,9 @@ Release 2.0.3-alpha - Unreleased
|
|||
YARN-372. Move InlineDispatcher from hadoop-yarn-server-resourcemanager to
|
||||
hadoop-yarn-common (sseth via hitesh)
|
||||
|
||||
YARN-370. Fix SchedulerUtils to correctly round up the resource for
|
||||
containers. (Zhijie Shen via acmurthy)
|
||||
|
||||
Release 2.0.2-alpha - 2012-09-07
|
||||
|
||||
YARN-9. Rename YARN_HOME to HADOOP_YARN_HOME. (vinodkv via acmurthy)
|
||||
|
|
|
@ -108,7 +108,8 @@ public class SchedulerUtils {
|
|||
Resource normalized =
|
||||
Resources.normalize(
|
||||
resourceCalculator, ask.getCapability(), minimumResource);
|
||||
ask.setCapability(normalized);
|
||||
ask.getCapability().setMemory(normalized.getMemory());
|
||||
ask.getCapability().setVirtualCores(normalized.getVirtualCores());
|
||||
}
|
||||
|
||||
}
|
||||
|
|
|
@ -19,6 +19,7 @@
|
|||
package org.apache.hadoop.yarn.server.resourcemanager.scheduler;
|
||||
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
|
||||
import org.apache.hadoop.yarn.api.records.Resource;
|
||||
import org.apache.hadoop.yarn.api.records.ResourceRequest;
|
||||
|
@ -42,23 +43,35 @@ public class TestSchedulerUtils {
|
|||
|
||||
// case negative memory
|
||||
ask.setCapability(Resources.createResource(-1024));
|
||||
Resource before = ask.getCapability();
|
||||
SchedulerUtils.normalizeRequest(ask, resourceCalculator, null, minResource);
|
||||
Resource after = ask.getCapability();
|
||||
assertEquals(minMemory, ask.getCapability().getMemory());
|
||||
assertTrue(before == after);
|
||||
|
||||
// case zero memory
|
||||
ask.setCapability(Resources.createResource(0));
|
||||
before = ask.getCapability();
|
||||
SchedulerUtils.normalizeRequest(ask, resourceCalculator, null, minResource);
|
||||
after = ask.getCapability();
|
||||
assertEquals(minMemory, ask.getCapability().getMemory());
|
||||
assertTrue(before == after);
|
||||
|
||||
// case memory is a multiple of minMemory
|
||||
ask.setCapability(Resources.createResource(2 * minMemory));
|
||||
before = ask.getCapability();
|
||||
SchedulerUtils.normalizeRequest(ask, resourceCalculator, null, minResource);
|
||||
after = ask.getCapability();
|
||||
assertEquals(2 * minMemory, ask.getCapability().getMemory());
|
||||
assertTrue(before == after);
|
||||
|
||||
// case memory is not a multiple of minMemory
|
||||
ask.setCapability(Resources.createResource(minMemory + 10));
|
||||
before = ask.getCapability();
|
||||
SchedulerUtils.normalizeRequest(ask, resourceCalculator, null, minResource);
|
||||
after = ask.getCapability();
|
||||
assertEquals(2 * minMemory, ask.getCapability().getMemory());
|
||||
assertTrue(before == after);
|
||||
|
||||
}
|
||||
|
||||
|
@ -73,24 +86,33 @@ public class TestSchedulerUtils {
|
|||
|
||||
// case negative memory/vcores
|
||||
ask.setCapability(Resources.createResource(-1024, -1));
|
||||
Resource before = ask.getCapability();
|
||||
SchedulerUtils.normalizeRequest(
|
||||
ask, resourceCalculator, clusterResource, minResource);
|
||||
Resource after = ask.getCapability();
|
||||
assertEquals(minResource, ask.getCapability());
|
||||
assertTrue(before == after);
|
||||
|
||||
// case zero memory/vcores
|
||||
ask.setCapability(Resources.createResource(0, 0));
|
||||
before = ask.getCapability();
|
||||
SchedulerUtils.normalizeRequest(
|
||||
ask, resourceCalculator, clusterResource, minResource);
|
||||
after = ask.getCapability();
|
||||
assertEquals(minResource, ask.getCapability());
|
||||
assertEquals(1, ask.getCapability().getVirtualCores());
|
||||
assertEquals(1024, ask.getCapability().getMemory());
|
||||
assertTrue(before == after);
|
||||
|
||||
// case non-zero memory & zero cores
|
||||
ask.setCapability(Resources.createResource(1536, 0));
|
||||
before = ask.getCapability();
|
||||
SchedulerUtils.normalizeRequest(
|
||||
ask, resourceCalculator, clusterResource, minResource);
|
||||
after = ask.getCapability();
|
||||
assertEquals(Resources.createResource(2048, 1), ask.getCapability());
|
||||
assertEquals(1, ask.getCapability().getVirtualCores());
|
||||
assertEquals(2048, ask.getCapability().getMemory());
|
||||
assertTrue(before == after);
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue