Mercurial > hg > Papers > 2020 > soto-midterm

comparison src/iterateCall.cbc @ 1:73127e0ab57c

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
(none)
author soto@cr.ie.u-ryukyu.ac.jp
date Tue, 08 Sep 2020 18:38:08 +0900
parents
children
comparison
equal deleted inserted replaced
0:b919985837a3 1:73127e0ab57c
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 }