void smpi_set_host_pstate_(integer *pstate_index)
{
smpi_set_host_pstate(static_cast<int>(*pstate_index));
}
int main(int argc, char *argv[])
{
int rank, pstates;
int i;
char buf[1024];
char *s;
size_t sz, x;
int err;
err = MPI_Init(&argc, &argv);
if (err != MPI_SUCCESS) {
fprintf(stderr, "MPI_init failed: %d\n", err);
exit(EXIT_FAILURE);
}
err = MPI_Comm_rank(MPI_COMM_WORLD, &rank); /* Get id of this process */
if (err != MPI_SUCCESS) {
fprintf(stderr, "MPI_Comm_rank failed: %d", err);
MPI_Abort(MPI_COMM_WORLD, EXIT_FAILURE);
exit(EXIT_FAILURE);
}
pstates = smpi_get_host_nb_pstates();
s = buf;
sz = sizeof buf;
x = snprintf(s, sz,
"[%.6f] [rank %d] Pstates: %d; Powers: %.0f",
MPI_Wtime(), rank, pstates, smpi_get_host_power_peak_at(0));
if (x < sz) {
s += x;
sz -= x;
} else
sz = 0;
for (i = 1; i < pstates; i++) {
x = snprintf(s, sz, ", %.0f", smpi_get_host_power_peak_at(i));
if (x < sz) {
s += x;
sz -= x;
} else
sz = 0;
}
fprintf(stderr, "%s%s\n", buf, (sz ? "" : " [...]"));
for (i = 0; i < pstates; i++) {
smpi_set_host_pstate(i);
fprintf(stderr, "[%.6f] [rank %d] Current pstate: %d; Current power: %.0f\n",
MPI_Wtime(), rank, i, smpi_get_host_current_power_peak());
SMPI_SAMPLE_FLOPS(1e9) {
/* imagine here some code running for 1e9 flops... */
}
fprintf(stderr, "[%.6f] [rank %d] Energy consumed: %g Joules.\n",
MPI_Wtime(), rank, smpi_get_host_consumed_energy());
}
err = MPI_Finalize();
if (err != MPI_SUCCESS) {
fprintf(stderr, "MPI_Finalize failed: %d\n", err);
MPI_Abort(MPI_COMM_WORLD, EXIT_FAILURE);
exit(EXIT_FAILURE);
}
return EXIT_SUCCESS;
}
void smpi_set_host_pstate_(integer *pstate_index)
{
smpi_set_host_pstate((int)*pstate_index);
}
