diff options
| -rw-r--r-- | mbw.c | 429 |
1 files changed, 221 insertions, 208 deletions
@@ -1,3 +1,6 @@ +/* + * vim: ai ts=4 sts=4 sw=4 cinoptions=>4 expandtab + */ #define _GNU_SOURCE #include <stdio.h> @@ -21,57 +24,62 @@ #define DEFAULT_BLOCK_SIZE 262144 /* - * Bare-bones memory bandwidth tester + * MBW memory bandwidth benchmark + * + * 2006, 2012 Andras.Horvath@gmail.com + * + * http://github.com/raas/mbw * - * Andras.Horvath@cern.ch * compile with: * gcc -O -o mbw mbw.c * * run with eg.: * * ./mbw 300 - * + * * or './mbw -h' for help * * watch out for swap usage (or turn off swap) */ -void usage() { - printf("Usage: mbw [options] array_size_in_MiB\n"); - printf("Options:\n"); - printf(" -n: number of runs per test\n"); - printf(" -a: Don't display average\n"); - printf(" -t0: memcpy test\n"); - printf(" -t1: dumb (b[i]=a[i] style) test\n"); - printf(" -t2 : memcpy test with fixed block size\n"); - printf(" -b <size>: block size in bytes for -t2 (default: %d)\n",DEFAULT_BLOCK_SIZE); - printf(" -q: quiet (print statistics only)\n"); - printf("(will then use two arrays, watch out for swapping)\n"); - printf("'Bandwidth' is amount of data copied over the time this operation took.\n"); - printf("\nThe default is to run all tests available.\n"); +void usage() +{ + printf("Usage: mbw [options] array_size_in_MiB\n"); + printf("Options:\n"); + printf(" -n: number of runs per test\n"); + printf(" -a: Don't display average\n"); + printf(" -t0: memcpy test\n"); + printf(" -t1: dumb (b[i]=a[i] style) test\n"); + printf(" -t2 : memcpy test with fixed block size\n"); + printf(" -b <size>: block size in bytes for -t2 (default: %d)\n", DEFAULT_BLOCK_SIZE); + printf(" -q: quiet (print statistics only)\n"); + printf("(will then use two arrays, watch out for swapping)\n"); + printf("'Bandwidth' is amount of data copied over the time this operation took.\n"); + printf("\nThe default is to run all tests available.\n"); } /* ------------------------------------------------------ */ /* allocate a test array and fill it with data * so as to force Linux to _really_ allocate it */ -long *make_array(unsigned long long asize) { - unsigned long long t; - unsigned int long_size=sizeof(long); - long *a; - - a=calloc(asize, long_size); - - if(NULL==a) { - perror("Error allocating memory"); - exit(1); - } - - /* make sure both arrays are allocated, fill with pattern */ - for(t=0; t<asize; t++) { - a[t]=0xaa; - } - return a; +long *make_array(unsigned long long asize) +{ + unsigned long long t; + unsigned int long_size=sizeof(long); + long *a; + + a=calloc(asize, long_size); + + if(NULL==a) { + perror("Error allocating memory"); + exit(1); + } + + /* make sure both arrays are allocated, fill with pattern */ + for(t=0; t<asize; t++) { + a[t]=0xaa; + } + return a; } /* actual benchmark */ @@ -81,40 +89,41 @@ long *make_array(unsigned long long asize) { * * return value: elapsed time in seconds */ -double worker(unsigned long long asize, long *a, long *b, int type, unsigned long long block_size) { - unsigned long long t; - struct timeval starttime,endtime; - double te; - unsigned int long_size=sizeof(long); - /* array size in bytes */ - unsigned long long array_bytes=asize*long_size; - - if(type==1) { /* memcpy test */ - /* timer starts */ - gettimeofday(&starttime, NULL); - memcpy(b,a,array_bytes); - /* timer stops */ - gettimeofday(&endtime, NULL); - } else if(type==2) { /* memcpy block test */ - gettimeofday(&starttime, NULL); - for(t=0; t<array_bytes; t+=block_size) { - b=mempcpy(b,a,block_size); - } - if(t>array_bytes){ - b=mempcpy(b,a,t-array_bytes); - } - gettimeofday(&endtime, NULL); - } else { /* dumb test */ - gettimeofday(&starttime, NULL); - for(t=0; t<asize; t++) { - b[t]=a[t]; - } - gettimeofday(&endtime, NULL); - } - - te=((double)(endtime.tv_sec*1000000-starttime.tv_sec*1000000+endtime.tv_usec-starttime.tv_usec))/1000000; - - return te; +double worker(unsigned long long asize, long *a, long *b, int type, unsigned long long block_size) +{ + unsigned long long t; + struct timeval starttime, endtime; + double te; + unsigned int long_size=sizeof(long); + /* array size in bytes */ + unsigned long long array_bytes=asize*long_size; + + if(type==1) { /* memcpy test */ + /* timer starts */ + gettimeofday(&starttime, NULL); + memcpy(b, a, array_bytes); + /* timer stops */ + gettimeofday(&endtime, NULL); + } else if(type==2) { /* memcpy block test */ + gettimeofday(&starttime, NULL); + for(t=0; t<array_bytes; t+=block_size) { + b=mempcpy(b, a, block_size); + } + if(t>array_bytes) { + b=mempcpy(b, a, t-array_bytes); + } + gettimeofday(&endtime, NULL); + } else { /* dumb test */ + gettimeofday(&starttime, NULL); + for(t=0; t<asize; t++) { + b[t]=a[t]; + } + gettimeofday(&endtime, NULL); + } + + te=((double)(endtime.tv_sec*1000000-starttime.tv_sec*1000000+endtime.tv_usec-starttime.tv_usec))/1000000; + + return te; } /* ------------------------------------------------------ */ @@ -126,149 +135,153 @@ double worker(unsigned long long asize, long *a, long *b, int type, unsigned lon * * return value: - */ -void printout(double te, double mt, int type) { - switch(type) { - case 0: - printf("Method: MEMCPY\t"); - break; - case 1: printf("Method: DUMB\t"); - break; - case 2: printf("Method: MCBLOCK\t"); - break; - } - printf("Elapsed: %.5f\t",te); - printf("MiB: %.5f\t",mt); - printf("Copy: %.3f MiB/s\n", mt/te); - return; +void printout(double te, double mt, int type) +{ + switch(type) { + case 0: + printf("Method: MEMCPY\t"); + break; + case 1: + printf("Method: DUMB\t"); + break; + case 2: + printf("Method: MCBLOCK\t"); + break; + } + printf("Elapsed: %.5f\t", te); + printf("MiB: %.5f\t", mt); + printf("Copy: %.3f MiB/s\n", mt/te); + return; } /* ------------------------------------------------------ */ -int main(int argc, char **argv) { - unsigned int long_size=0; - double te,te_sum; /* time elapsed */ - unsigned long long asize=0; /* array size (elements in array) */ - int i; - long *a, *b; /* the two arrays to be copied from/to */ - int o; /* getopt options */ - unsigned long testno; - - /* options */ - int nr_loops=DEFAULT_NR_LOOPS; /* how many runs to average? */ - unsigned long long block_size=DEFAULT_BLOCK_SIZE; /* fixed memcpy block size for -t2 */ - int showavg=1; /* show average, -a */ - /* what tests to run (-t x) */ - int tests[MAX_TESTS]; - double mt=0; /* MiBytes transferred == array size in MiB */ - int quiet=0; /* suppress extra messages */ - - tests[0]=0; - tests[1]=0; - tests[2]=0; - - while((o=getopt(argc, argv, "haqn:t:b:")) != EOF) { - switch(o) { - case 'h': - usage(); - exit(1); - break; - case 'a': /* suppress printing average */ - showavg=0; - break; - case 'n': /* no. loops */ - nr_loops=strtoul(optarg, (char **)NULL, 10); - break; - case 't': /* test to run */ - testno=strtoul(optarg, (char **)NULL, 10); - if(0>testno) { - printf("Error: test number must be between 0 and %d\n",MAX_TESTS); - usage(); - exit(1); - } - tests[testno]=1; - break; - case 'b': /* block size in bytes*/ - block_size=strtoull(optarg, (char **)NULL, 10); - if(0>=block_size) { - printf("Error: what block size do you mean?\n"); - usage(); - exit(1); - } - break; - case 'q': /* quiet */ - quiet=1; - break; - default: - break; - } - } - - /* default is to run all tests if no specific tests were requested */ - if( (tests[0]+tests[1]+tests[2]) == 0) { - tests[0]=1; - tests[1]=1; - tests[2]=1; - } - - if(optind<argc) { - mt=strtoul(argv[optind++], (char **)NULL, 10); - } else { - printf("Error: no array size given!\n"); - usage(); - exit(1); - } - - if(0>=mt) { - printf("Error: array size wrong!\n"); - usage(); - exit(1); - } - - /* ------------------------------------------------------ */ - - long_size=sizeof(long); /* the size of long on this platform */ - asize=1024*1024/long_size*mt; /* how many longs then in one array? */ - - if(asize*long_size < block_size) { - printf("Error: array size larger than block size!\n"); - usage(); - exit(1); - } - - if(!quiet) { - printf("Long uses %d bytes. ",long_size); - printf("Allocating 2*%lld elements = %lld bytes of memory.\n",asize,2*asize*long_size); - if(tests[2]) { - printf("Using %lld bytes as blocks for memcpy block copy test.\n",block_size); - } - } - - a=make_array(asize); - b=make_array(asize); - - /* ------------------------------------------------------ */ - if(!quiet) { - printf("Getting down to business... Doing %d runs per test.\n",nr_loops); - } - - /* run all tests requested, the proper number of times */ - for(testno=0; testno<MAX_TESTS; testno++) { - te_sum=0; - if(tests[testno]) { - for (i=0; i<nr_loops; i++) { - te=worker(asize,a,b,testno,block_size); - te_sum+=te; - printf("%d\t",i); - printout(te,mt,testno); - } - if(showavg) { - printf("AVG\t"); - printout(te_sum/nr_loops,mt,testno); - } - } - } - - free(a); - free(b); - return 0; +int main(int argc, char **argv) +{ + unsigned int long_size=0; + double te, te_sum; /* time elapsed */ + unsigned long long asize=0; /* array size (elements in array) */ + int i; + long *a, *b; /* the two arrays to be copied from/to */ + int o; /* getopt options */ + unsigned long testno; + + /* options */ + + /* how many runs to average? */ + int nr_loops=DEFAULT_NR_LOOPS; + /* fixed memcpy block size for -t2 */ + unsigned long long block_size=DEFAULT_BLOCK_SIZE; + /* show average, -a */ + int showavg=1; + /* what tests to run (-t x) */ + int tests[MAX_TESTS]; + double mt=0; /* MiBytes transferred == array size in MiB */ + int quiet=0; /* suppress extra messages */ + + tests[0]=0; + tests[1]=0; + tests[2]=0; + + while((o=getopt(argc, argv, "haqn:t:b:")) != EOF) { + switch(o) { + case 'h': + usage(); + exit(1); + break; + case 'a': /* suppress printing average */ + showavg=0; + break; + case 'n': /* no. loops */ + nr_loops=strtoul(optarg, (char **)NULL, 10); + break; + case 't': /* test to run */ + testno=strtoul(optarg, (char **)NULL, 10); + if(0>testno) { + printf("Error: test number must be between 0 and %d\n", MAX_TESTS); + exit(1); + } + tests[testno]=1; + break; + case 'b': /* block size in bytes*/ + block_size=strtoull(optarg, (char **)NULL, 10); + if(0>=block_size) { + printf("Error: what block size do you mean?\n"); + exit(1); + } + break; + case 'q': /* quiet */ + quiet=1; + break; + default: + break; + } + } + + /* default is to run all tests if no specific tests were requested */ + if( (tests[0]+tests[1]+tests[2]) == 0) { + tests[0]=1; + tests[1]=1; + tests[2]=1; + } + + if(optind<argc) { + mt=strtoul(argv[optind++], (char **)NULL, 10); + } else { + printf("Error: no array size given!\n"); + exit(1); + } + + if(0>=mt) { + printf("Error: array size wrong!\n"); + exit(1); + } + + /* ------------------------------------------------------ */ + + long_size=sizeof(long); /* the size of long on this platform */ + asize=1024*1024/long_size*mt; /* how many longs then in one array? */ + + if(asize*long_size < block_size) { + printf("Error: array size larger than block size (%llu bytes)!\n", block_size); + exit(1); + } + + if(!quiet) { + printf("Long uses %d bytes. ", long_size); + printf("Allocating 2*%lld elements = %lld bytes of memory.\n", asize, 2*asize*long_size); + if(tests[2]) { + printf("Using %lld bytes as blocks for memcpy block copy test.\n", block_size); + } + } + + a=make_array(asize); + b=make_array(asize); + + /* ------------------------------------------------------ */ + if(!quiet) { + printf("Getting down to business... Doing %d runs per test.\n", nr_loops); + } + + /* run all tests requested, the proper number of times */ + for(testno=0; testno<MAX_TESTS; testno++) { + te_sum=0; + if(tests[testno]) { + for (i=0; i<nr_loops; i++) { + te=worker(asize, a, b, testno, block_size); + te_sum+=te; + printf("%d\t", i); + printout(te, mt, testno); + } + if(showavg) { + printf("AVG\t"); + printout(te_sum/nr_loops, mt, testno); + } + } + } + + free(a); + free(b); + return 0; } + |