Mercurial > hg > Members > kono > Cerium
annotate TaskManager/Cell/CellTaskManagerImpl.cc @ 638:671fca057ad3
hmmmm
author | Shinji KONO <kono@ie.u-ryukyu.ac.jp> |
---|---|
date | Thu, 19 Nov 2009 18:18:20 +0900 |
parents | 5530fa36d42e |
children | a909c50081c2 |
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" |
12 | |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
13 CellTaskManagerImpl::~CellTaskManagerImpl() |
67 | 14 { |
109 | 15 delete speThreads; |
16 delete [] speTaskList; | |
17 delete [] speTaskList_bg; | |
18 /** | |
321 | 19 * bufferManager は |
20 * ppeManager のなかで delete してもらう | |
109 | 21 */ |
22 // delete bufferManager; | |
23 delete [] flag_sendTaskList; | |
24 | |
25 delete ppeManager; | |
67 | 26 } |
27 | |
57 | 28 void |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
29 CellTaskManagerImpl::init() |
57 | 30 { |
637 | 31 spe_running = 0; |
476
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
32 taskListImpl = new CellTaskListInfo; |
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
33 taskListImpl->init(machineNum*2); |
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
34 |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
35 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
|
36 |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
37 htaskImpl = activeTaskQueue ; // any HTaskInfo |
65 | 38 |
39 speThreads = new SpeThreads(machineNum); | |
40 speThreads->init(); | |
109 | 41 |
42 speTaskList = new TaskListPtr[machineNum]; | |
43 speTaskList_bg = new TaskListPtr[machineNum]; | |
476
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
44 |
109 | 45 for (int i = 0; i < machineNum; i++) { |
476
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
46 speTaskList[i] = taskListImpl->create(); |
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
47 speTaskList_bg[i] = taskListImpl->create(); |
109 | 48 } |
49 | |
50 flag_sendTaskList = new int[machineNum]; | |
51 for (int i = 0; i < machineNum; i++) { | |
52 flag_sendTaskList[i] = 1; | |
476
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
53 } |
321 | 54 // PPE 側の管理をする Manager |
109 | 55 ppeManager = new FifoTaskManagerImpl(machineNum); |
479
5bda98b0b56d
Double Linked List base TaskQueue
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
476
diff
changeset
|
56 // 大半のTaskQueueInfoは、共有される |
637 | 57 MainScheduler *mscheduler = new MainScheduler; |
58 ppeManager->init(mscheduler, this); | |
619 | 59 |
60 ppeManager->get_scheduler()->set_manager(this); | |
637 | 61 |
619 | 62 schedTaskManager = new SchedTask(); |
63 schedTaskManager->init(0,0,0,ppeManager->get_scheduler()); | |
57 | 64 } |
65 | |
109 | 66 void |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
67 CellTaskManagerImpl::append_activeTask(HTaskPtr task) |
109 | 68 { |
69 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
|
70 ppeManager->append_activeTask(task); |
109 | 71 } else { |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
72 activeTaskQueue->addLast(task); |
109 | 73 } |
74 } | |
75 | |
321 | 76 // SPE_ANY が指定されていた時に |
77 // これをインクリメントしつつ呼ぶことにする。 | |
78 // 乱数使ってもいいけどさ。 | |
109 | 79 int cur_anySpeid = 0; |
80 | |
81 /** | |
321 | 82 * ActiveTaskQueue から Task を |
83 * 各 SPE に渡す TaskList に入れる | |
109 | 84 * |
321 | 85 * ここの activeTaskQueue は FifoTaskManagerImpl のと意味が違い、 |
86 * spe に渡される Task だけ入っている | |
109 | 87 */ |
88 void | |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
89 CellTaskManagerImpl::set_runTaskList() |
70 | 90 { |
321 | 91 // ここ...直すかな |
70 | 92 TaskListPtr list; |
481
4896dffad67c
Double linked list modification done (tested on Mac OS X)
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
479
diff
changeset
|
93 |
70 | 94 TaskPtr task; |
109 | 95 int speid; |
70 | 96 |
501
4a2c9ff53605
Cell inData/outData DMA removal
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
496
diff
changeset
|
97 while (HTaskPtr htask = activeTaskQueue->poll()) { |
70 | 98 |
109 | 99 if (htask->cpu_type == SPE_ANY) { |
100 speid = cur_anySpeid++; | |
101 cur_anySpeid = (cur_anySpeid < machineNum) | |
102 ? cur_anySpeid : 0; | |
70 | 103 } else { |
321 | 104 // -1 してるのは |
105 // htask->cpu_type - CPU_SPE で | |
106 // SPE0 = 1, SPE1 = 2, ... SPE5 = 6 ってなってるので | |
107 // 配列的 (SPE0 = arr[0], SPE1 = arr[1]) にするため | |
109 | 108 speid = htask->cpu_type - CPU_SPE - 1; |
109 | |
321 | 110 // SPU の数以上が指定されていれば |
111 // とりあえず MAX_USE_SPE_NUM (実際に動く SPE の最大数) で | |
112 // あまり求めてそれを使うことにする。 | |
113 // ここで判定するもんでもないか? | |
109 | 114 if (speid >= machineNum) { |
115 speid %= machineNum; | |
116 } | |
70 | 117 } |
109 | 118 |
119 list = speTaskList_bg[speid]; | |
120 | |
121 if (list->length >= TASK_MAX_SIZE) { | |
476
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
122 TaskListPtr newList = taskListImpl->create(); |
109 | 123 newList = TaskListInfo::append(newList, speTaskList_bg[speid]); |
124 speTaskList_bg[speid] = newList; | |
125 list = newList; | |
126 } | |
127 | |
128 task = &list->tasks[list->length++]; | |
629
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
129 TaskPtr stask = (TaskPtr) task; |
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
130 *stask = *(TaskPtr) htask; |
70 | 131 } |
109 | 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; | |
138 MailQueuePtr ppeMail = NULL; | |
70 | 139 |
321 | 140 // PPE 側で動く TaskList です |
141 // FifoTaskManagerImpl::run と上手く合うように | |
142 // こんなことやってますよ | |
637 | 143 // |
144 // 本来は、別 thread で動かすべきだろう... | |
109 | 145 ppeTaskList = ppeManager->get_runTaskList(); |
146 if (!ppeTaskList) { | |
147 goto cont; | |
148 } | |
70 | 149 |
321 | 150 // SPE からの Mailbox Check は |
151 // PPE 側の schedule から抜けて来たときに行う | |
152 // (speThreads で Blocking Mailbox read と | |
153 // セマフォとか使ってやってもいいが、今はこの方式で) | |
637 | 154 // |
155 // すべてのspe task が finish task を待つ場合は、ppeTaskList の | |
156 // 判定だけで十分だが、そうでない場合は、spe の task が残っているか | |
157 // どうかを調べる必要がある。 | |
158 // | |
109 | 159 do { |
160 ppeMail = ppeManager->schedule(ppeTaskList); | |
161 cont: | |
162 ppeTaskList = mail_check(ppeMail); | |
637 | 163 } while (ppeTaskList || spe_running >0); |
70 | 164 } |
165 | |
65 | 166 /** |
321 | 167 * SPE からのメールをチェックする |
109 | 168 * |
169 * @param [mail_list] | |
321 | 170 * PPE 側で動く Scheduler からのメールリスト |
171 * この中で PPE 側の mail check も行う | |
109 | 172 * |
321 | 173 * @return PPE Scheduler に対してのメール。 |
174 * 次に実行する TaskList のアドレスや、終了コマンドを送る | |
109 | 175 */TaskListPtr |
65 | 176 CellTaskManagerImpl::mail_check(MailQueuePtr mail_list) |
57 | 177 { |
321 | 178 // PPE Scheduler からの mail check |
485 | 179 ppeManager->mail_check(mail_list, waitTaskQueue); |
65 | 180 |
70 | 181 do { |
629
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
182 memaddr data; |
109 | 183 |
321 | 184 // SPE Scheduler からの mail check |
244 | 185 for (int id = 0; id < machineNum; id++) { |
186 while (speThreads->check_mail(id, 1, &data)) { | |
109 | 187 /** |
321 | 188 * MY_SPE_STATUS_READY: SPE が持ってた Task 全て終了 |
189 * MY_SPE_NOP: 特に意味のないコマンド | |
190 * それ以外:終了したタスク(PPEにあるのでアドレス | |
109 | 191 * |
321 | 192 * MY_SPE_NOP が 0 なので、 |
193 * 下のように data > MY_SPE_NOP とかしています。 | |
194 * 一目でよくわからない書き方なんで、直したいところですが。。。 | |
109 | 195 */ |
321 | 196 // 名前あとでちゃんと決めよう => MY_SPE_... とかじゃなくて |
629
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
197 if (data == (memaddr)MY_SPE_STATUS_READY) { |
334
20f2459041cb
[in test_render] push L key , call show_dma_wait, but incomplete.
e065746@localhost.localdomain
parents:
321
diff
changeset
|
198 //__debug_ppe("[SPE %d] finish\n", id); |
109 | 199 flag_sendTaskList[id] = 1; |
637 | 200 spe_running--; |
629
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
201 } else if (data == (memaddr)MY_SPE_COMMAND_MALLOC) { |
334
20f2459041cb
[in test_render] push L key , call show_dma_wait, but incomplete.
e065746@localhost.localdomain
parents:
321
diff
changeset
|
202 //__debug_ppe("[PPE] MALLOC COMMAND from [SPE %d]\n", id); |
109 | 203 |
204 /** | |
321 | 205 * info[0] = alloc_id; (CellScheduler::mainMem_alloc 参照) |
109 | 206 * info[1] = alloc_addr; |
207 */ | |
629
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
208 memaddr alloc_info[2]; |
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
209 long alloc_size; |
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
210 long command; |
109 | 211 |
244 | 212 speThreads->get_mail(id, 2, alloc_info); |
629
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
213 command = (long)alloc_info[0]; |
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
214 alloc_size = (long)alloc_info[1]; |
244 | 215 |
109 | 216 |
629
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
217 alloc_info[1] = (memaddr)allocate(alloc_size); |
244 | 218 /* |
321 | 219 * allocate された領域は今の SPE buffer にリンクとして接続する |
220 * ここでは TaskList を allocate(new) して登録してやろうか | |
244 | 221 */ |
109 | 222 |
629
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
223 //__debug_ppe("[PPE] MALLOCED 0x%lx from [SPE %d]\n", alloc_info[1],id); |
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
224 // 今のところ何もしてない。どうも、この allocate を free |
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
225 // するのは、SPE task が返した値を見て行うらしい。それは、 |
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
226 // 忘れやすいのではないか? |
300
0691f38195f6
remove TaskManager/Fifo/MainScheduler.o TaskManeger/Test/test_render/test_nogl
tkaito@localhost.localdomain
parents:
298
diff
changeset
|
227 speThreads->add_output_tasklist(command, alloc_info[1], alloc_size); |
273 | 228 |
621 | 229 speThreads->send_mail(id, 2, alloc_info); |
629
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
230 } else if (data > (memaddr)MY_SPE_NOP) { |
334
20f2459041cb
[in test_render] push L key , call show_dma_wait, but incomplete.
e065746@localhost.localdomain
parents:
321
diff
changeset
|
231 //__debug_ppe("[PPE] recv from [SPE %d] : 0x%x\n", id, data); |
619 | 232 HTaskPtr task = (HTaskPtr)data; |
233 task->post_func(schedTaskManager, task->post_arg1, task->post_arg2); | |
234 check_task_finish(task); | |
70 | 235 } |
65 | 236 } |
237 } | |
109 | 238 |
321 | 239 // 依存関係を満たしたタスクをアクティブに |
109 | 240 wakeup_waitTask(); |
241 | |
321 | 242 // SPE に送る TaskList の準備 |
109 | 243 set_runTaskList(); |
244 | |
321 | 245 // TaskList 待ちの SPE に TaskList を送る |
109 | 246 for (int i = 0; i < machineNum; i++) { |
247 if (flag_sendTaskList[i] == 1 && speTaskList_bg[i]->length >= 1 ) { | |
248 send_taskList(i); | |
637 | 249 spe_running++; |
109 | 250 } |
251 } | |
70 | 252 |
321 | 253 // 現在の FifoTaskManager の仕様では |
254 // ・PPE で実行するタスクが無くなれば終了する | |
255 // であり、この場合もし SPE にタスクが残っていても | |
256 // メインループから抜けてプログラム終了となってしまうので | |
257 // ここでストップかけてます。 | |
481
4896dffad67c
Double linked list modification done (tested on Mac OS X)
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
479
diff
changeset
|
258 } while (ppeManager->activeTaskQueue->empty() && !waitTaskQueue->empty()); |
109 | 259 |
260 return ppeManager->get_runTaskList(); | |
261 } | |
66 | 262 |
109 | 263 /** |
321 | 264 * 条件を満たしたら SPE に TaskList を送信する |
265 * 条件1. SPE が持ってた TaskList を終了して、次の TaskList を待ってる | |
266 * 条件2. SPE に送る TaskList に Task がある | |
109 | 267 * |
321 | 268 * SPE で実行終了した TaskList [speTaskList] と |
269 * これから実行する TaskList [speTaskList_bg] のバッファを入れ替える | |
270 * ついでに実行終了したやつは clear しておく。 | |
109 | 271 */ |
272 void | |
273 CellTaskManagerImpl::send_taskList(int id) | |
274 { | |
275 TaskListPtr tmp; | |
276 | |
277 tmp = speTaskList[id]; | |
278 speTaskList[id] = speTaskList_bg[id]; | |
279 speTaskList_bg[id] = tmp; | |
280 | |
476
926d6aff886c
CellBufferManager removed
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
380
diff
changeset
|
281 taskListImpl->clear_taskList(speTaskList_bg[id]); |
109 | 282 |
629
8843edf37c0e
Cell 64 bit tried, but not yet worked.
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
621
diff
changeset
|
283 speThreads->send_mail(id, 1, (memaddr *)&speTaskList[id]); |
109 | 284 flag_sendTaskList[id] = 0; |
57 | 285 } |
286 | |
637 | 287 #if 0 |
109 | 288 void* |
506 | 289 CellTaskManagerImpl::allocate(int size, int alignment) |
290 { | |
635
8cc609285bbe
SimpleTask worked on Mac OS X
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
629
diff
changeset
|
291 if (size==0) return 0; |
506 | 292 void *buff; |
293 posix_memalign(&buff, alignment, size); | |
294 return buff; | |
295 } | |
296 | |
297 void* | |
109 | 298 CellTaskManagerImpl::allocate(int size) |
299 { | |
635
8cc609285bbe
SimpleTask worked on Mac OS X
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents:
629
diff
changeset
|
300 if (size==0) return 0; |
109 | 301 void *buff; |
302 | |
303 posix_memalign(&buff, DEFAULT_ALIGNMENT, size); | |
244 | 304 |
321 | 305 // bzero はコストが高いのでやりたくない |
506 | 306 // bzero(buff, size); |
109 | 307 |
308 return buff; | |
309 } | |
310 | |
380 | 311 Scheduler* |
312 CellTaskManagerImpl::get_scheduler() | |
313 { | |
314 return ppeManager->get_scheduler(); | |
315 } | |
637 | 316 #endif |
380 | 317 |
109 | 318 #ifdef __CERIUM_CELL__ |
57 | 319 TaskManagerImpl* |
320 create_impl(int num) | |
321 { | |
67 | 322 return new CellTaskManagerImpl(num); |
57 | 323 } |
109 | 324 #endif // __CERIUM_CELL |