Mercurial > hg > Members > kono > Cerium
annotate TaskManager/Cell/CellTaskManagerImpl.cc @ 801:974cd68383b3
TaslListInfo
author | Shinji KONO <kono@ie.u-ryukyu.ac.jp> |
---|---|
date | Sat, 22 May 2010 18:20:16 +0900 |
parents | 5088d70e66c5 |
children | 8a6f1fa038de |
rev | line source |
---|---|
321 | 1 #define DEBUG |
2 #include "error.h" | |
57 | 3 #include <stdio.h> |
4 #include <stdlib.h> | |
5 #include <string.h> | |
6 #include "CellTaskManagerImpl.h" | |
476
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
7 #include "CellTaskListInfo.h" |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
8 #include "HTaskInfo.h" |
621 | 9 #include "SchedTask.h" |
635
8cc609285bbe
SimpleTask worked on Mac OS X
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
629
diff
changeset
|
10 #include "MainScheduler.h" |
57 | 11 #include "types.h" |
672 | 12 #include "SysFunc.h" |
13 | |
720 | 14 static void send_alloc_reply(CellTaskManagerImpl *tm, int id, SpeThreads *speThreads); |
57 | 15 |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
16 CellTaskManagerImpl::~CellTaskManagerImpl() |
67 | 17 { |
109 | 18 delete speThreads; |
19 delete [] speTaskList; | |
20 delete [] speTaskList_bg; | |
21 /** | |
321 | 22 * bufferManager は |
23 * ppeManager のなかで delete してもらう | |
109 | 24 */ |
25 // delete bufferManager; | |
26 delete [] flag_sendTaskList; | |
27 | |
28 delete ppeManager; | |
67 | 29 } |
30 | |
57 | 31 void |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
32 CellTaskManagerImpl::init() |
57 | 33 { |
637 | 34 spe_running = 0; |
476
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
35 taskListImpl = new CellTaskListInfo; |
664 | 36 taskListImpl->init(16); |
476
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
37 |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
38 activeTaskQueue = new HTaskInfo(); |
481
4896dffad67c
Double linked list modification done (tested on Mac OS X)
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
479
diff
changeset
|
39 |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
40 htaskImpl = activeTaskQueue ; // any HTaskInfo |
65 | 41 |
42 speThreads = new SpeThreads(machineNum); | |
43 speThreads->init(); | |
109 | 44 |
45 speTaskList = new TaskListPtr[machineNum]; | |
46 speTaskList_bg = new TaskListPtr[machineNum]; | |
476
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
47 |
109 | 48 for (int i = 0; i < machineNum; i++) { |
476
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
49 speTaskList[i] = taskListImpl->create(); |
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
50 speTaskList_bg[i] = taskListImpl->create(); |
109 | 51 } |
52 | |
53 flag_sendTaskList = new int[machineNum]; | |
54 for (int i = 0; i < machineNum; i++) { | |
55 flag_sendTaskList[i] = 1; | |
476
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
56 } |
321 | 57 // PPE 側の管理をする Manager |
109 | 58 ppeManager = new FifoTaskManagerImpl(machineNum); |
479
5bda98b0b56d
Double Linked List base TaskQueue
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
476
diff
changeset
|
59 // 大半のTaskQueueInfoは、共有される |
637 | 60 MainScheduler *mscheduler = new MainScheduler; |
61 ppeManager->init(mscheduler, this); | |
619 | 62 |
63 ppeManager->get_scheduler()->set_manager(this); | |
637 | 64 |
619 | 65 schedTaskManager = new SchedTask(); |
66 schedTaskManager->init(0,0,0,ppeManager->get_scheduler()); | |
57 | 67 } |
68 | |
109 | 69 void |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
70 CellTaskManagerImpl::append_activeTask(HTaskPtr task) |
109 | 71 { |
72 if (task->cpu_type == CPU_PPE) { | |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
73 ppeManager->append_activeTask(task); |
109 | 74 } else { |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
75 activeTaskQueue->addLast(task); |
109 | 76 } |
77 } | |
78 | |
321 | 79 // SPE_ANY が指定されていた時に |
80 // これをインクリメントしつつ呼ぶことにする。 | |
81 // 乱数使ってもいいけどさ。 | |
109 | 82 int cur_anySpeid = 0; |
83 | |
84 /** | |
321 | 85 * ActiveTaskQueue から Task を |
86 * 各 SPE に渡す TaskList に入れる | |
109 | 87 * |
321 | 88 * ここの activeTaskQueue は FifoTaskManagerImpl のと意味が違い、 |
89 * spe に渡される Task だけ入っている | |
109 | 90 */ |
91 void | |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
92 CellTaskManagerImpl::set_runTaskList() |
70 | 93 { |
109 | 94 int speid; |
70 | 95 |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
96 while (HTaskPtr htask = activeTaskQueue->poll()) { |
70 | 97 |
109 | 98 if (htask->cpu_type == SPE_ANY) { |
99 speid = cur_anySpeid++; | |
100 cur_anySpeid = (cur_anySpeid < machineNum) | |
101 ? cur_anySpeid : 0; | |
70 | 102 } else { |
321 | 103 // -1 してるのは |
104 // htask->cpu_type - CPU_SPE で | |
105 // SPE0 = 1, SPE1 = 2, ... SPE5 = 6 ってなってるので | |
106 // 配列的 (SPE0 = arr[0], SPE1 = arr[1]) にするため | |
109 | 107 speid = htask->cpu_type - CPU_SPE - 1; |
108 | |
321 | 109 // SPU の数以上が指定されていれば |
110 // とりあえず MAX_USE_SPE_NUM (実際に動く SPE の最大数) で | |
111 // あまり求めてそれを使うことにする。 | |
112 // ここで判定するもんでもないか? | |
109 | 113 if (speid >= machineNum) { |
114 speid %= machineNum; | |
115 } | |
70 | 116 } |
109 | 117 |
801 | 118 TaskListInfoPtr list = speTaskList_bg[speid]; |
119 set_taskList(htask, list); | |
70 | 120 } |
109 | 121 } |
122 | |
123 void | |
722 | 124 CellTaskManagerImpl::sendTaskList() |
721 | 125 { |
126 for (int i = 0; i < machineNum; i++) { | |
127 if (flag_sendTaskList[i] == 1 && speTaskList_bg[i]->length >= 1 ) { | |
128 send_taskList(i); | |
129 spe_running++; | |
130 } | |
131 } | |
132 } | |
133 | |
134 void | |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
135 CellTaskManagerImpl::run() |
109 | 136 { |
137 TaskListPtr ppeTaskList = NULL; | |
70 | 138 |
109 | 139 do { |
719
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
140 // PPE side |
801 | 141 ppeTaskList = ppeManager->set_runTaskList(); |
721 | 142 if (ppeTaskList) |
143 ppeManager->sendTaskList(ppeTaskList); | |
720 | 144 ppeManager->mail_check(); |
719
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
145 // SPE side |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
146 do { |
720 | 147 mail_check(); |
719
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
148 // SPE に送る TaskList の準備 |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
149 set_runTaskList(); |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
150 // TaskList 待ちの SPE に TaskList を送る |
722 | 151 sendTaskList(); |
152 } while (ppeManager->activeTaskQueue->empty() && spe_running >0 ); | |
721 | 153 } while (ppeTaskList || spe_running >0); |
154 if (!waitTaskQueue->empty()) { | |
736 | 155 get_scheduler()->printf("Dead lock detected\n"); |
721 | 156 } |
70 | 157 } |
158 | |
65 | 159 /** |
321 | 160 * SPE からのメールをチェックする |
640
a909c50081c2
SimpeTask on Cell worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
638
diff
changeset
|
161 */ |
a909c50081c2
SimpeTask on Cell worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
638
diff
changeset
|
162 |
719
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
163 void |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
164 CellTaskManagerImpl::mail_check() |
57 | 165 { |
719
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
166 memaddr data; |
109 | 167 |
719
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
168 // SPE Scheduler からの mail check |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
169 for (int id = 0; id < machineNum; id++) { |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
170 while (speThreads->has_mail(id, 1, &data)) { |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
171 if (data == (memaddr)MY_SPE_STATUS_READY) { |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
172 // MY_SPE_STATUS_READY: SPE が持ってた Task 全て終了 |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
173 flag_sendTaskList[id] = 1; |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
174 spe_running--; |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
175 } else if (data == (memaddr)MY_SPE_COMMAND_MALLOC) { |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
176 // MY_SPE_COMMAND_MALLOC SPE からのmain memory request |
720 | 177 send_alloc_reply(this, id, speThreads); |
719
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
178 } else if (data > (memaddr)MY_SPE_NOP) { |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
179 // 終了したタスク(PPEにあるのでアドレス) |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
180 HTaskPtr task = (HTaskPtr)data; |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
181 task->post_func(schedTaskManager, task->post_arg1, task->post_arg2); |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
182 check_task_finish(task,waitTaskQueue); |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
183 } |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
184 // MY_SPE_NOP: 特に意味のないコマンド |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
185 } |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
186 } |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
187 } |
109 | 188 |
719
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
189 static void |
720 | 190 send_alloc_reply(CellTaskManagerImpl *tm, int id, SpeThreads *speThreads) |
719
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
191 { |
273 | 192 |
719
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
193 /** |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
194 * info[0] = alloc_id; (CellScheduler::mainMem_alloc 参照) |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
195 * info[1] = alloc_addr; |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
196 */ |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
197 memaddr alloc_info[2]; |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
198 long alloc_size; |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
199 long command; |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
200 |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
201 speThreads->get_mail(id, 2, alloc_info); |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
202 command = (long)alloc_info[0]; |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
203 alloc_size = (long)alloc_info[1]; |
109 | 204 |
719
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
205 |
720 | 206 alloc_info[1] = (memaddr)tm->allocate(alloc_size); |
719
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
207 //__debug_ppe("[PPE] MALLOCED 0x%lx from [SPE %d]\n", alloc_info[1],id); |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
208 // 今のところ何もしてない。どうも、この allocate を free |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
209 // するのは、SPE task が返した値を見て行うらしい。それは、 |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
210 // 忘れやすいのではないか? |
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
211 speThreads->add_output_tasklist(command, alloc_info[1], alloc_size); |
109 | 212 |
719
cafffff0f45a
clean up scheduler main loop
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
718
diff
changeset
|
213 speThreads->send_mail(id, 2, alloc_info); |
109 | 214 } |
66 | 215 |
109 | 216 /** |
321 | 217 * 条件を満たしたら SPE に TaskList を送信する |
218 * 条件1. SPE が持ってた TaskList を終了して、次の TaskList を待ってる | |
219 * 条件2. SPE に送る TaskList に Task がある | |
109 | 220 * |
321 | 221 * SPE で実行終了した TaskList [speTaskList] と |
222 * これから実行する TaskList [speTaskList_bg] のバッファを入れ替える | |
223 * ついでに実行終了したやつは clear しておく。 | |
109 | 224 */ |
225 void | |
226 CellTaskManagerImpl::send_taskList(int id) | |
227 { | |
228 TaskListPtr tmp; | |
229 | |
230 tmp = speTaskList[id]; | |
231 speTaskList[id] = speTaskList_bg[id]; | |
232 speTaskList_bg[id] = tmp; | |
233 | |
476
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
234 taskListImpl->clear_taskList(speTaskList_bg[id]); |
109 | 235 |
629
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
236 speThreads->send_mail(id, 1, (memaddr *)&speTaskList[id]); |
57 | 237 } |
238 | |
672 | 239 void CellTaskManagerImpl::show_profile() { |
240 for (int id = 0; id < machineNum; id++) { | |
241 HTaskPtr t = create_task(ShowTime); | |
242 t->set_cpu((CPU_TYPE)(id+2)); | |
243 t->spawn(); | |
244 } | |
245 } | |
246 | |
247 void CellTaskManagerImpl::start_profile() { | |
248 for (int id = 0; id < machineNum; id++) { | |
249 HTaskPtr t = create_task(StartProfile); | |
250 t->set_cpu((CPU_TYPE)(id+2)); | |
251 t->spawn(); | |
252 } | |
253 } | |
254 | |
380 | 255 |
109 | 256 #ifdef __CERIUM_CELL__ |
57 | 257 TaskManagerImpl* |
258 create_impl(int num) | |
259 { | |
67 | 260 return new CellTaskManagerImpl(num); |
57 | 261 } |
109 | 262 #endif // __CERIUM_CELL |