githubmirror/numactl

stream_main.c

36 строк731 B

Andi Kleen

Import numactl 0.9

23 июл 2014, 20:24

23 июл 2014, 20:24f4e30c1

#include <stdio.h>
#include <sys/mman.h>
#include <stdlib.h>
#include "numa.h"
#include "numaif.h"
#include "util.h"
#include "stream_lib.h"

void usage(void)
{
	exit(1);
}

char *policy = "default";

/* Run STREAM with a numa policy */
int main(int ac, char **av)
{
	nodemask_t nodes; 
	char *map;
	long size;
	int policy;
	policy = parse_policy(av[1], av[2]); 
	nodemask_zero(&nodes);
	if (av[1] && av[2])
		nodes = nodemask(av[2]);
	size = stream_memsize();  
	map = mmap(NULL, size, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS,
		   0, 0); 
	if (map == (char*)-1) exit(1); 
	if (mbind(map, size, policy, &nodes.n[0], NUMA_NUM_NODES, 0) < 0) 
		perror("mbind"), exit(1);
	stream_init(map); 
	stream_test(NULL);
	return 0;
}