Skip to content

Commit c33a698

Browse files
committed
Update slurm memory logging
1 parent 9c0f8c2 commit c33a698

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

cluster/src/org/labkey/cluster/pipeline/SlurmExecutionEngine.java

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,15 +14,20 @@
1414
import org.labkey.api.data.ContainerManager;
1515
import org.labkey.api.pipeline.PipelineJob;
1616
import org.labkey.api.pipeline.PipelineJobException;
17+
import org.labkey.api.pipeline.PipelineJobService;
18+
import org.labkey.api.pipeline.PipelineService;
19+
import org.labkey.api.pipeline.PipelineStatusFile;
1720
import org.labkey.api.util.FileUtil;
1821
import org.labkey.api.util.Pair;
22+
import org.labkey.api.writer.PrintWriters;
1923
import org.labkey.cluster.ClusterManager;
2024
import org.labkey.cluster.ClusterServiceImpl;
2125
import org.quartz.JobExecutionException;
2226

2327
import java.io.File;
2428
import java.io.FileWriter;
2529
import java.io.IOException;
30+
import java.io.PrintWriter;
2631
import java.util.ArrayList;
2732
import java.util.Arrays;
2833
import java.util.HashMap;
@@ -311,6 +316,15 @@ else if (headerFound)
311316
if (bytes > requestInBytes)
312317
{
313318
info = "Job exceeded memory, max was: " + FileSizeFormatter.convertBytesToUnit(bytes, 'G') + "G";
319+
320+
PipelineStatusFile sf = PipelineService.get().getStatusFile(job.getJobId());
321+
if (sf != null)
322+
{
323+
try (PrintWriter writer = PrintWriters.getPrintWriter(new File(sf.getFilePath())))
324+
{
325+
writer.println(info + ". Raw slurm value: " + maxRSS);
326+
}
327+
}
314328
}
315329
}
316330
}

0 commit comments

Comments
 (0)