1
|
1 #include "../context.h"
|
|
2 #interface "Iterator.h"
|
|
3 #interface "TaskManager.h"
|
|
4 #include <stdio.h>
|
|
5
|
|
6 Iterator* createMultiDimIterator(struct Context* context, int x, int y, int z) {
|
|
7 struct Iterator* iterator = new Iterator();
|
|
8 struct MultiDimIterator* multiDimIterator = new MultiDimIterator();
|
|
9 iterator->iterator = (union Data*)multiDimIterator;
|
|
10 iterator->exec = C_execMultiDimIterator;
|
|
11 iterator->barrier = C_barrierMultiDimIterator;
|
|
12 multiDimIterator->x = x;
|
|
13 multiDimIterator->y = y;
|
|
14 multiDimIterator->z = z;
|
|
15 multiDimIterator->count = x * y * z;
|
|
16 multiDimIterator->counterX = 0;
|
|
17 multiDimIterator->counterY = 0;
|
|
18 multiDimIterator->counterZ = 0;
|
|
19 return iterator;
|
|
20 }
|
|
21
|
|
22 /**
|
|
23 * create iterateTask with index, that copy from task argument
|
|
24 * @return created iterateTask
|
|
25 * @param task task of the copy source
|
|
26 * @x index
|
|
27 */
|
|
28 struct Context* createMultiDimIterateTask(struct Context* task, int x, int y, int z) {
|
|
29 struct Context* task1 = NEW(struct Context);
|
|
30 initContext(task1);
|
|
31 task1->taskManager = task->taskManager;
|
|
32 task1->next = task->next;
|
|
33 task1->iterate = 1;
|
|
34 task1->iterator = task->iterator;
|
|
35 task1->idgCount = task->idgCount;
|
|
36 task1->idg = task->idg;
|
|
37 task1->maxIdg = task->maxIdg;
|
|
38 for(int i = task1->idg; i < task1->maxIdg; i++) {
|
|
39 task1->data[i] = task->data[i];
|
|
40 }
|
|
41
|
|
42 // create index data gear and register input data to iterate task
|
|
43 struct MultiDim* multiDim = &ALLOCATE_DATA_GEAR(task1, MultiDim)->MultiDim;
|
|
44 multiDim->x = x;
|
|
45 multiDim->y = y;
|
|
46 multiDim->z = z;
|
|
47 task1->data[task1->maxIdg++] = (union Data*)multiDim;
|
|
48 task1->odg = task->odg + 1;
|
|
49 task1->maxOdg = task->maxOdg + 1;
|
|
50 for (int i = task1->odg; i < task1->maxOdg; i++) {
|
|
51 task1->data[i] = task->data[i-1];
|
|
52 }
|
|
53
|
|
54 return task1;
|
|
55 }
|
|
56
|
|
57 __code execMultiDimIterator(struct MultiDimIterator* iterator, struct Context* task, int numGPU, __code next(...)) {
|
|
58 // No GPU device
|
|
59 if (numGPU == 0) {
|
|
60 goto execMultiDimIterator1();
|
|
61 }
|
|
62 task->iterate = 1;
|
|
63 task->gpu = 1;
|
|
64 struct TaskManager* taskManager = task->taskManager;
|
|
65 goto taskManager->spawn(task, next(...));
|
|
66 }
|
|
67
|
|
68 __code execMultiDimIterator1(struct MultiDimIterator* iterator, struct Context* task, __code next(...)) {
|
|
69 int x = iterator->counterX;
|
|
70 int y = iterator->counterY;
|
|
71 int z = iterator->counterZ;
|
|
72 struct Context* iterateTask = createMultiDimIterateTask(task, x, y, z);
|
|
73 struct TaskManager* taskManager = task->taskManager;
|
|
74 goto taskManager->spawn(iterateTask, execMultiDimIterator2);
|
|
75 }
|
|
76
|
|
77 __code execMultiDimIterator2(struct MultiDimIterator* iterator, struct Context* task, __code next(...)) {
|
|
78 if (++iterator->counterX >= iterator->x) {
|
|
79 iterator->counterX = 0;
|
|
80 if (++iterator->counterY >= iterator->y) {
|
|
81 iterator->counterY = 0;
|
|
82 if (++iterator->counterZ >= iterator->z) {
|
|
83 iterator->counterZ = 0;
|
|
84 goto next(...);
|
|
85 }
|
|
86 }
|
|
87 }
|
|
88 goto execMultiDimIterator1();
|
|
89 }
|
|
90
|
|
91 __code barrierMultiDimIterator(struct MultiDimIterator* iterator, struct Context* task, __code next(...), __code whenWait(...)) {
|
|
92 if (task->gpu || __sync_fetch_and_sub(&iterator->count, 1) == 1) {
|
|
93 goto next(...);
|
|
94 }
|
|
95 goto whenWait(...);
|
|
96 }
|