memusage.C File Reference

#include "memusage.h"
#include "converse.h"
#include <stdio.h>
#include <sys/types.h>
#include <unistd.h>
#include <malloc.h>

Go to the source code of this file.

Functions
unsigned long	memusage_mstats ()
unsigned long	memusage_mallinfo ()
unsigned long	memusage_ps ()
unsigned long	memusage_proc_self_stat ()
unsigned long	memusage (const char **source)

Function Documentation

unsigned long memusage

(

const char **

source = 0

)

Copyright (c) 1995, 1996, 1997, 1998, 1999, 2000 by The Board of Trustees of the University of Illinois. All rights reserved.

Definition at line 158 of file memusage.C.

References memusage_mallinfo(), memusage_mstats(), memusage_proc_self_stat(), and memusage_ps().

Referenced by main::main(), memusage_kB(), and memusage_MB().

                                            {

  unsigned long memtotal = 0;
  const char* s = "ERROR";

  if ( ! CmiMemoryIs(CMI_MEMORY_IS_OS) ) {
    memtotal = CmiMemoryUsage();  s = "CmiMemoryUsage";
  }

#if CMK_BLUEGENEQ
  if( ! memtotal) { memtotal = memusage_bgq(); s="Kernel_GetMemorySize on BG/Q"; }
#endif

#if CMK_BLUEGENEP
  if( ! memtotal) { memtotal = memusage_bgp(); s="mallinfo on BG/P"; }
#endif

#if defined(WIN32) && !defined(__CYGWIN__)
  if ( ! memtotal ) {
    memtotal = CmiMemoryUsage();  s = "GetProcessMemoryInfo";
  }
#endif

  if ( ! memtotal ) {
    memtotal = memusage_proc_self_stat();  s = "/proc/self/stat";
  }

  if ( ! memtotal ) { memtotal = memusage_mstats(); s = "mstats"; }

  if ( ! memtotal ) { memtotal = memusage_mallinfo(); s = "mallinfo"; }

  if ( ! memtotal ) { memtotal = memusageinit::memusage_sbrk(); s = "sbrk"; }

  if ( ! memtotal ) { memtotal = memusage_ps(); s = "ps"; }

  if ( ! memtotal ) { memtotal = CmiMemoryUsage();  s = "CmiMemoryUsage"; }

  if ( ! memtotal ) s = "nothing";

  if ( source ) *source = s;

  return memtotal;

}

unsigned long memusage_mallinfo

(

)

Definition at line 68 of file memusage.C.

References if().

Referenced by memusage().

                                  {
  struct mallinfo mi = mallinfo();
  // unsigned long memtotal = mi.usmblks + mi.uordblks + mi.hblkhd;
  unsigned long memtotal = (unsigned int) mi.uordblks;
  unsigned long memtotal2 = (unsigned int) mi.usmblks;
  memtotal2 += (unsigned int) mi.hblkhd;
  if ( memtotal2 > memtotal ) memtotal = memtotal2;

  // printf("mallinfo %d %d %d\n", mi.usmblks, mi.uordblks, mi.hblkhd);

  return memtotal;
}

unsigned long memusage_mstats ( )

inline

Definition at line 60 of file memusage.C.

Referenced by memusage().

{ return 0; }

unsigned long memusage_proc_self_stat ( )

inline

Definition at line 131 of file memusage.C.

Referenced by memusage().

                                               {
#ifdef NO_PS
  return 0;
#else
  static int failed_once = 0;
  if ( failed_once ) return 0;  // no point in retrying

  FILE *f = fopen("/proc/self/stat","r");
  if ( ! f ) { failed_once = 1; return 0; }
  for ( int i=0; i<22; ++i ) fscanf(f,"%*s");
#if defined(NAMD_CUDA) || defined(NAMD_HIP) || defined(NAMD_MIC)
  // skip vss, next value is rss in pages
  fscanf(f,"%*s");
#endif
  unsigned long vsz = 0;  // should remain 0 on failure
  fscanf(f,"%lu",&vsz);
  fclose(f);
#if defined(NAMD_CUDA) || defined(NAMD_HIP) || defined(NAMD_MIC)
  vsz *= sysconf(_SC_PAGESIZE);
#endif
  if ( ! vsz ) failed_once = 1;
  // printf("/proc/self/stat reports %d MB\n", vsz/(1024*1024));
  return vsz;
#endif
}

unsigned long memusage_ps ( )

inline

Definition at line 86 of file memusage.C.

Referenced by memusage().

                                   {
#ifdef NO_PS
  return 0;
#else
  char pscmd[100];
#if defined(NAMD_CUDA) || defined(NAMD_HIP) || defined(NAMD_MIC)
  sprintf(pscmd, "/bin/ps -o rss= -p %d", getpid());
#else
  sprintf(pscmd, "/bin/ps -o vsz= -p %d", getpid());
#endif
  unsigned long vsz = 0;
  FILE *p = popen(pscmd, "r");
  if ( p ) {
    fscanf(p, "%ld", &vsz);
    pclose(p);
  }
  return ( vsz * (unsigned long) 1024 );
#endif
}