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