Changeset 22f928f for memlog.cpp

Timestamp:

05/31/15 09:46:11 (9 years ago)

Author:

Hal Finkel <hfinkel@…>

Branches:

master

Children:

4598848

Parents:

a7b97b9

git-author:

Hal Finkel <hfinkel@…> (05/31/15 09:46:11)

git-committer:

Hal Finkel <hfinkel@…> (05/31/15 09:46:11)

Message:

use the job id under batch systems; misc cleanups

File:

• memlog.cpp (modified) (2 diffs)

memlog.cpp

@@ -37 +37 @@
 //   -Wl,--wrap,malloc,--wrap,free,--wrap,realloc,--wrap,calloc,--wrap,memalign /path/to/memlog_s.o -lpthread -ldl
 
-FILE *log_file = NULL;
+FILE *log_file = 0;
 static pthread_mutex_t log_mutex = PTHREAD_MUTEX_INITIALIZER;
 
@@ -58 +58 @@
 #endif
 
+  // If we're running under a common batch system, add the job id to the output
+  // file names (add it as a prefix so that sorting the files will sort by job
+  // first).
+  char *job_id = 0;
+  const char *job_id_vars[] =
+    { "COBALT_JOBID", "PBS_JOBID", "SLURM_JOB_ID", "JOB_ID" };
+  for (int i = 0; i < sizeof(job_id_vars)/sizeof(job_id_vars[0]); ++i) {
+    job_id = getenv(job_id_vars[i]);
+    if (job_id)
+      break;
+  }
+
   char log_name[PATH_MAX+1];
-  snprintf(log_name, PATH_MAX+1, "%s.%d.memlog", u.nodename, id);
+  if (job_id)
+    snprintf(log_name, PATH_MAX+1, "%s.%s.%d.memlog", job_id, u.nodename, id);
+  else
+    snprintf(log_name, PATH_MAX+1, "%s.%d.memlog", u.nodename, id);
   log_file = fopen(log_name, "w");
   if (!log_file)