codes-workload.c 5.92 KB
Newer Older
1
2
3
4
5
6
/*
 * Copyright (C) 2013 University of Chicago.
 * See COPYRIGHT notice in top-level directory.
 *
 */

7
#include <assert.h>
8
9
10
11
12

#include "ross.h"
#include "codes/codes-workload.h"
#include "codes-workload-method.h"

13
/* list of available methods.  These are statically compiled for now, but we
14
15
 * could make generators optional via autoconf tests etc. if needed
 */
16
extern struct codes_workload_method test_workload_method;
17
extern struct codes_workload_method bgp_io_workload_method;
18
#ifdef USE_DARSHAN
19
extern struct codes_workload_method darshan_io_workload_method;
20
#endif
21
22
23
#ifdef USE_RECORDER
extern struct codes_workload_method recorder_io_workload_method;
#endif
24

25
static struct codes_workload_method *method_array[] =
26
27
28
29
30
{
    &test_workload_method,
    &bgp_io_workload_method,
#ifdef USE_DARSHAN
    &darshan_io_workload_method,
31
32
33
#endif
#ifdef USE_RECORDER
    &recorder_io_workload_method,
34
35
#endif
    NULL};
36

Philip Carns's avatar
Philip Carns committed
37
38
39
40
41
/* This shim layer is responsible for queueing up reversed operations and
 * re-issuing them so that the underlying workload generator method doesn't
 * have to worry about reverse events.
 *
 * NOTE: we could make this faster with a smarter data structure.  For now
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
 * we just have a linked list of rank_queue structs, one per rank that has
 * opened the workload.  We then have a linked list off of each of those
 * to hold a lifo queue of operations that have been reversed for that rank.
 */

/* holds an operation that has been reversed */
struct rc_op
{
    struct codes_workload_op op;
    struct rc_op* next;
};

/* tracks lifo queue of reversed operations for a given rank */
struct rank_queue
{
    int rank;
    struct rc_op *lifo;
    struct rank_queue *next;
};

static struct rank_queue *ranks = NULL;

64
65
int codes_workload_load(const char* type, const char* params, int rank)
{
66
67
    int i;
    int ret;
68
    struct rank_queue *tmp;
69
70
71
72
73

    for(i=0; method_array[i] != NULL; i++)
    {
        if(strcmp(method_array[i]->method_name, type) == 0)
        {
74
            /* load appropriate workload generator */
75
76
77
78
79
            ret = method_array[i]->codes_workload_load(params, rank);
            if(ret < 0)
            {
                return(-1);
            }
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98

            /* are we tracking information for this rank yet? */
            tmp = ranks;
            while(tmp)
            {
                if(tmp->rank == rank)
                    break;
                tmp = tmp->next;
            }
            if(tmp == NULL)
            {
                tmp = malloc(sizeof(*tmp));
                assert(tmp);
                tmp->rank = rank;
                tmp->lifo = NULL;
                tmp->next = ranks;
                ranks = tmp;
            }

99
100
101
102
103
            return(i);
        }
    }

    fprintf(stderr, "Error: failed to find workload generator %s\n", type);
104
105
106
    return(-1);
}

107
void codes_workload_get_next(int wkld_id, int rank, struct codes_workload_op *op)
108
{
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
    struct rank_queue *tmp;
    struct rc_op *tmp_op;

    /* first look to see if we have a reversed operation that we can
     * re-issue
     */
    tmp = ranks;
    while(tmp)
    {
        if(tmp->rank == rank)
            break;
        tmp = tmp->next;
    }
    assert(tmp);
    if(tmp->lifo)
    {
        tmp_op = tmp->lifo;
126
        tmp->lifo = tmp_op->next;
127
128
129

        *op = tmp_op->op;
        free(tmp_op);
130
        //printf("codes_workload_get_next re-issuing reversed operation.\n");
131
132
133
134
        return;
    }

    /* ask generator for the next operation */
135
    //printf("codes_workload_get_next issuing new operation.\n");
136
137
    method_array[wkld_id]->codes_workload_get_next(rank, op);

138
139
140
    return;
}

141
void codes_workload_get_next_rc(int wkld_id, int rank, const struct codes_workload_op *op)
142
{
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
    struct rank_queue *tmp;
    struct rc_op *tmp_op;

    tmp = ranks;
    while(tmp)
    {
        if(tmp->rank == rank)
            break;
        tmp = tmp->next;
    }
    assert(tmp);

    tmp_op = malloc(sizeof(*tmp_op));
    assert(tmp_op);
    tmp_op->op = *op;
    tmp_op->next = tmp->lifo;
    tmp->lifo = tmp_op;
160

161
162
163
    return;
}

164
int codes_workload_get_rank_cnt(const char* type, const char* params)
165
{
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
    int i;
    int rank_cnt;

    for(i=0; method_array[i] != NULL; i++)
    {
        if(strcmp(method_array[i]->method_name, type) == 0)
        {
            rank_cnt = method_array[i]->codes_workload_get_rank_cnt(params);
            assert(rank_cnt > 0);
            return(rank_cnt);
        }
    }

    fprintf(stderr, "Error: failed to find workload generator %s\n", type);
    return(-1);
181
182
}

183
184
185
void codes_workload_print_op(FILE *f, struct codes_workload_op *op, int rank){
    switch(op->op_type){
        case CODES_WK_END:
186
            fprintf(f, "op: rank:%d type:end\n", rank);
187
188
            break;
        case CODES_WK_DELAY:
189
            fprintf(f, "op: rank:%d type:delay seconds:%lf\n",
190
191
192
                    rank, op->u.delay.seconds);
            break;
        case CODES_WK_BARRIER:
193
            fprintf(f, "op: rank:%d type:barrier count:%d root:%d\n",
194
195
196
                    rank, op->u.barrier.count, op->u.barrier.root);
            break;
        case CODES_WK_OPEN:
197
            fprintf(f, "op: rank:%d type:open file_id:%lu flag:%d\n",
198
199
200
                    rank, op->u.open.file_id, op->u.open.create_flag);
            break;
        case CODES_WK_CLOSE:
201
            fprintf(f, "op: rank:%d type:close file_id:%lu\n",
202
203
204
                    rank, op->u.close.file_id);
            break;
        case CODES_WK_WRITE:
205
206
            fprintf(f, "op: rank:%d type:write "
                       "file_id:%lu off:%lu size:%lu\n",
207
                    rank, op->u.write.file_id, op->u.write.offset,
208
209
210
                    op->u.write.size);
            break;
        case CODES_WK_READ:
211
212
            fprintf(f, "op: rank:%d type:read "
                       "file_id:%lu off:%lu size:%lu\n",
213
                    rank, op->u.read.file_id, op->u.read.offset,
214
215
216
217
218
                    op->u.read.size);
            break;
    }
}

219
220
221
222
223
224
225
226
/*
 * Local variables:
 *  c-indent-level: 4
 *  c-basic-offset: 4
 * End:
 *
 * vim: ft=c ts=8 sts=4 sw=4 expandtab
 */