numam/util/memloadgen.cc

#include "gen.hh"
#include <cstdlib>
#include "ntr.h"
#include <getopt.h>
#include <unistd.h>
#include <topo.h>

static void
usage()
{
	fprintf(stdout,
	    "Usage:\n"
		"    -v: verbose mode\n"
	    "    -b: MLG bytes per second\n"
	    "    -x: MLG thread affinity mask\n"
	    "    -X: MLG target domain affinity mask\n"
		"    -S: shared buffer\n"
		"    -i: iterations\n");
	fflush(stdout);
}

int main(int argc, char * argv[])
{
	ntr_init();
	ntr_set_level(NTR_DEP_USER1, NTR_LEVEL_WARNING);

	size_t arr_sz = 0;
	uint32_t iter = -1;
	cpuset_t threads;
	int shared_buffer = 0;
	cpuset_t domain_mask;
	{
		int c;
		// parse arguments
		while ((c = getopt(argc, argv, "hb:X:x:vi:S")) != -1) {
			switch (c) {
			case 'v':
				ntr_set_level(NTR_DEP_USER1,
				    ntr_get_level(NTR_DEP_USER1) + 1);
				break;
			case 'h':
				usage();
				exit(0);
			case 'b':
				arr_sz = strtoull(optarg, nullptr, 10);
				break;
			case 'i':
				iter = strtoul(optarg, nullptr, 10);
				break;
			case 'X':
				cpulist_to_cpuset(optarg, &domain_mask);
				break;
			case 'x':
				cpulist_to_cpuset(optarg, &threads);
				break;
			case 'S':
				shared_buffer = 1;
				break;
			default:
				usage();
				exit(0);
			}
		}
	}

	ntr(NTR_DEP_USER1, NTR_LEVEL_INFO, "MLG: [size: %ld iter: %u, # threads: 0x%d, domain: 0x%ld]\n", arr_sz, iter, CPU_COUNT(&threads), CPU_FFS(&domain_mask) - 1);

	// init topo
	if (topo_init(ntr_get_level(NTR_DEP_USER1) - NTR_LEVEL_WARNING) != 0) {
		fprintf(stderr, "libtopo init failed!\n");
		exit(1);
	}

	bool success = false;
	memload_generator::memload_generator_options opts;
	opts.chunk_size = arr_sz;
	opts.iteration = iter;
	opts.shared_buffer = shared_buffer;
	opts.verbose = 1;

    auto mgen = new memload_generator(&threads, &domain_mask, &opts, &success);
	while(!mgen->check_done()) {
		usleep(10000);
	}

	unsigned long bps = mgen->get_bps();
	ntr(NTR_DEP_USER1, NTR_LEVEL_INFO,
			"main: MLG bps = %ld ~= %ldM\n", bps, bps / 1024 / 1024);

	delete mgen;

	return 0;
}