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