6 #include "model-assert.h"
9 static thrd_t *threads;
10 static unsigned int *input;
11 static unsigned int *output;
12 static int num_threads;
16 thrd_t curr = thrd_current();
18 for (i = 0; i < num_threads; i++)
19 if (curr.priv == threads[i].priv)
27 unsigned int idx1, idx2, idx3;
28 unsigned int reclaimNode;
31 static void main_task(void *param)
34 int pid = *((int *)param);
35 if (pid % procs == 0) {
37 succ1 = dequeue(queue, &idx1, &reclaimNode);
39 } else if (pid % procs == 1) {
41 succ2 = dequeue(queue, &idx2, &reclaimNode);
45 int user_main(int argc, char **argv)
49 unsigned int in_sum = 0, out_sum = 0;
52 queue = (queue_t*) calloc(1, sizeof(*queue));
56 threads = (thrd_t*) malloc(num_threads * sizeof(thrd_t));
57 param = (int*) malloc(num_threads * sizeof(*param));
58 input = (unsigned int *) calloc(num_threads, sizeof(*input));
59 output = (unsigned int *) calloc(num_threads, sizeof(*output));
61 atomic_init(&x[0], 0);
62 atomic_init(&x[1], 0);
63 atomic_init(&x[2], 0);
64 init_queue(queue, num_threads);
65 for (i = 0; i < num_threads; i++) {
67 thrd_create(&threads[i], main_task, ¶m[i]);
69 for (i = 0; i < num_threads; i++)
70 thrd_join(threads[i]);
72 for (i = 0; i < num_threads; i++) {
76 for (i = 0; i < num_threads; i++)
77 printf("input[%d] = %u\n", i, input[i]);
78 for (i = 0; i < num_threads; i++)
79 printf("output[%d] = %u\n", i, output[i]);
81 MODEL_ASSERT(in_sum == out_sum);