1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
#include <stdio.h>
#include <stdlib.h>
#include "IEEEIO.hh"
#include "vatoi.hh"
#include <sys/time.h>
double MPI_Wtime(){
timeval tv;
gettimeofday(&tv,0);
return tv.tv_sec + tv.tv_usec/1000000.0l;
}
class CommandLine {
public:
int dims[3];
char *programname;
int bufsize;
int ntrials;
void setDefaults(){
dims[0]=dims[1]=dims[2]=128;
bufsize=0;
ntrials=40;
}
CommandLine(int argc,char *argv[]){
setDefaults();
parse(argc,argv);
}
void parse(int argc,char *argv[]){
programname=argv[0];
for(int i=1;i<argc;i++){
char *flag = argv[i],*val=argv[i+1];
if(*flag=='-') flag++;
else continue; // ignore non-flags
if(!strcmp(flag,"dims")){
if(++i<argc) sscanf(val,"%u,%u,%u",dims,dims+1,dims+2);
}
else if(!strcmp(flag,"buffer")) {
if(++i<argc)
bufsize = vatoi(val);
}
else if(*flag=='n') {
ntrials=atoi(val);
i++;
}
else if(*flag=='h') usage();
}
}
void usage(){
printf("Usage: %s <-dims int,int,int> -n <ntrials>\n",
programname);
printf("\tdims: Set the dimensions of the dataset.\n");
printf("\tn: Set number of trials of writing the dataset.\n");
printf("\tbuffer: Set buffer cache size in bytes.\n");
}
void printStatus(FILE *f=stdout){
fprintf(f,"%s: dims=%u,%u,%u\n",
programname,
dims[0],dims[1],dims[2]);
}
};
int main(int argc,char *argv[]){
CommandLine cmdln(argc,argv);
int size=IObase::nElements(3,cmdln.dims);
int i;
float *data = new float[size];
for(i=0;i<size;i++) data[i] = (float)i;
IEEEIO *file = new IEEEIO("rawdata.raw",IObase::Write);
if(cmdln.bufsize>0) file->bufferOn(cmdln.bufsize);
double stime = MPI_Wtime();
for(i=0;i<cmdln.ntrials;i++){ // do this 40 times
file->reserveStream(IObase::Float32,3,cmdln.dims);
file->writeStream(data,size);
}
if(file) delete file; // make sure the file is deleted
double etime = MPI_Wtime();
printf("Elapsed time to write=%lf\n",etime-stime);
printf("IO performance = %f Megabytes/sec\n",
(float)(IObase::nBytes(IObase::Float32,3,cmdln.dims))*(float)(cmdln.ntrials)/(1024*1024*(etime-stime)));
delete data;
}
|
