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