1. Jeff Squyres
  2. mpi3-fortran

Source

mpi3-fortran / ompi / contrib / vt / vt / vtlib / vt_getcpu.c

/**
 * VampirTrace
 * http://www.tu-dresden.de/zih/vampirtrace
 *
 * Copyright (c) 2005-2012, ZIH, TU Dresden, Federal Republic of Germany
 *
 * Copyright (c) 1998-2005, Forschungszentrum Juelich, Juelich Supercomputing
 *                          Centre, Federal Republic of Germany
 *
 * See the file COPYING in the package base directory for details
 **/

#define _GNU_SOURCE

#include "config.h"

#include <errno.h>
#include <sched.h>
#include <string.h>

#include "vt_defs.h"
#include "vt_error.h"
#include "vt_getcpu.h"
#include "vt_trc.h"

/* counter id */
uint32_t vt_getcpu_cid = 0;

void vt_getcpu_init()
{
  vt_libassert(vt_misc_cgid != 0);

  /* write counter definition */
  vt_getcpu_cid = vt_def_counter(VT_CURRENT_THREAD, "CPU_ID", "#",
                                 VT_CNTR_ABS | VT_CNTR_NEXT,
                                 vt_misc_cgid, 0);
}

void vt_getcpu_finalize()
{
}

void vt_getcpu_read(uint32_t* value, uint8_t* changed)
{
  int cpuid;
  *changed = 0;

  if ( (cpuid = sched_getcpu()) == -1 )
    vt_error_msg("sched_getcpu: %s", strerror(errno));

  if( (uint32_t)cpuid != *value )
  {
    *value = (uint32_t)cpuid;
    *changed = 1;
  }
}