diff --git a/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 b/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 index 2496ac851ce..5534b74f6b0 100644 --- a/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 +++ b/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 @@ -21,8 +21,8 @@ import com.google.common.annotations.VisibleForTesting; import com.google.common.collect.ImmutableSet; import com.google.common.collect.Sets; -import org.apache.commons.logging.Log; -import org.apache.commons.logging.LogFactory; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; import org.apache.hadoop.util.StringUtils; import org.apache.hadoop.yarn.api.records.ContainerId; import org.apache.hadoop.yarn.api.records.Resource; @@ -52,7 +52,7 @@ * Allocate GPU resources according to requirements */ public class GpuResourceAllocator { - final static Log LOG = LogFactory.getLog(GpuResourceAllocator.class); + static final Logger LOG = LoggerFactory.getLogger(GpuResourceAllocator.class); private static final int WAIT_MS_PER_LOOP = 1000; private Set allowedGpuDevices = new TreeSet<>(); @@ -118,6 +118,8 @@ public synchronized void recoverAssignedGpus(ContainerId containerId) + containerId); } + LOG.info("Starting recovery of GpuDevice for {}.", containerId); + for (Serializable gpuDeviceSerializable : c.getResourceMappings() .getAssignedResources(GPU_URI)) { if (!(gpuDeviceSerializable instanceof GpuDevice)) { @@ -145,7 +147,11 @@ public synchronized void recoverAssignedGpus(ContainerId containerId) } usedDevices.put(gpuDevice, containerId); + LOG.info("ContainerId {} is assigned to GPU device {} on recovery.", + containerId, gpuDevice); } + + LOG.info("Finishing recovery of GpuDevice for {}.", containerId); } /**