Apache/hadoop
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

YARN-9099. GpuResourceAllocator#getReleasingGpus calculates number of GPUs in a wrong way. Contributed by Szilard Nemeth.

Browse Source 
(cherry picked from commit 71c49fa60faad2504b0411979a6e46e595b97a85)
This commit is contained in:
Sunil G 2019-01-31 09:25:29 +05:30
parent 4052b7ee60
commit 0e7060a1d5
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-nodemanager/src/main/java/org/apache/hadoop/yarn/server/nodemanager/containermanager/linux/resources/gpu/GpuResourceAllocator.java
View File

@ -258,10 +258,7 @@ private synchronized GpuAllocation internalAssignGpus(Container container)
private synchronized long getReleasingGpus() { private synchronized long getReleasingGpus() {
long releasingGpus = 0; long releasingGpus = 0;
Iterator<Map.Entry<GpuDevice, ContainerId>> iter = usedDevices.entrySet() for (ContainerId containerId : ImmutableSet.copyOf(usedDevices.values())) {
.iterator();
while (iter.hasNext()) {
ContainerId containerId = iter.next().getValue();
Container container; Container container;
if ((container = nmContext.getContainers().get(containerId)) != null) { if ((container = nmContext.getContainers().get(containerId)) != null) {
if (container.isContainerInFinalStates()) { if (container.isContainerInFinalStates()) {