Changeset a7b97b9

Timestamp:

05/30/15 23:54:09 (9 years ago)

Author:

Hal Finkel <hfinkel@…>

Branches:

master

Children:

22f928f

Parents:

518793d

git-author:

Hal Finkel <hfinkel@…> (05/30/15 23:54:09)

git-committer:

Hal Finkel <hfinkel@…> (05/30/15 23:54:09)

Message:

use kernel rank on the bgq instead of pid

Files:

• Makefile (modified) (1 diff)
• memlog.cpp (modified) (2 diffs)

Makefile

@@ -2 +2 @@
 CXXFLAGS = -std=gnu++0x -O3 -g
 
-CPPFLAGS =
+# When compiling with CXX=powerpc64-bgq-linux-g++, we need these:
+CPPFLAGS = -I/bgsys/drivers/ppcfloor -I/bgsys/drivers/ppcfloor/spi/include/kernel/cnk
+
 LDFLAGS = -lpthread -ldl
 

memlog.cpp

@@ -27 +27 @@
 #include <dlfcn.h>
 
+#ifdef __bgq__
+#include <spi/include/kernel/location.h>
+#endif
+
 using namespace std;
 
@@ -46 +50 @@
   uname(&u);
 
+  int id = (int) getpid();
+#ifdef __bgq__
+  // If we're really running on a BG/Q compute node, use the job rank instead
+  // of the pid because the node name might not really be globally unique.
+  if (!strcmp(u.sysname, "CNK") && !strcmp(u.machine, "BGQ"))
+    id = (int) Kernel_GetRank();
+#endif
+
   char log_name[PATH_MAX+1];
-  snprintf(log_name, PATH_MAX+1, "%s.%d.memlog", u.nodename, getpid());
+  snprintf(log_name, PATH_MAX+1, "%s.%d.memlog", u.nodename, id);
   log_file = fopen(log_name, "w");
   if (!log_file)