big-data/hadoop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

YARN-9370. Added logging for recovering assigned GPU devices.

Browse Source 
Contributed by Yesha Vora
This commit is contained in:
Eric Yang 2019-03-20 19:12:19 -04:00
parent f2b862cac6
commit 506502bb83
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
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

@ -119,6 +119,7 @@ public synchronized void recoverAssignedGpus(ContainerId containerId)
+ containerId); + containerId);
} }
LOG.info("Starting recovery of GpuDevice for {}.", containerId);
for (Serializable gpuDeviceSerializable : c.getResourceMappings() for (Serializable gpuDeviceSerializable : c.getResourceMappings()
.getAssignedResources(GPU_URI)) { .getAssignedResources(GPU_URI)) {
if (!(gpuDeviceSerializable instanceof GpuDevice)) { if (!(gpuDeviceSerializable instanceof GpuDevice)) {
@ -146,7 +147,10 @@ public synchronized void recoverAssignedGpus(ContainerId containerId)
} }
usedDevices.put(gpuDevice, containerId); usedDevices.put(gpuDevice, containerId);
LOG.info("ContainerId {} is assigned to GpuDevice {} on recovery.",
containerId, gpuDevice);
} }
LOG.info("Finished recovery of GpuDevice for {}.", containerId);
} }
/** /**