Mercurial > hg > Members > Moririn
annotate src/parallel_execution/worker.c @ 244:d1567718f12c
Fix error
author | Tatsuki IHA <e125716@ie.u-ryukyu.ac.jp> |
---|---|
date | Wed, 25 Jan 2017 01:48:47 +0900 |
parents | 9f3f8ed6ed9f |
children | ce262b2c1daf |
rev | line source |
---|---|
95
3e28ee215c0e
modify twice, use OSAtomiceCompareAndSwap
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
1 #include <libkern/OSAtomic.h> |
3e28ee215c0e
modify twice, use OSAtomiceCompareAndSwap
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
2 |
3e28ee215c0e
modify twice, use OSAtomiceCompareAndSwap
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
3 #include "context.h" |
3e28ee215c0e
modify twice, use OSAtomiceCompareAndSwap
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
4 #include "origin_cs.h" |
3e28ee215c0e
modify twice, use OSAtomiceCompareAndSwap
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
5 |
235 | 6 static void start_worker(Worker* worker); |
233 | 7 |
236 | 8 union Data* createCPUWorker(struct Context* context, int id, Queue* queue) { |
233 | 9 struct Worker* worker = ALLOC(context, Worker); |
10 struct CPUWorker* cpuWorker = ALLOC(context, CPUWorker); | |
232 | 11 worker->worker = (union Data*)cpuWorker; |
244 | 12 worker->tasks = queue; |
232 | 13 cpuWorker->id = id; |
222
77faa28128b4
Add taskSend for TaskManager
Tatsuki IHA <e125716@ie.u-ryukyu.ac.jp>
parents:
217
diff
changeset
|
14 worker->taskReceive = C_taskReceiveWorker; |
77faa28128b4
Add taskSend for TaskManager
Tatsuki IHA <e125716@ie.u-ryukyu.ac.jp>
parents:
217
diff
changeset
|
15 worker->shutdown = C_shutdownWorker; |
235 | 16 pthread_create(&worker->worker->CPUWorker.thread, NULL, (void*)&start_worker, worker); |
183 | 17 return (union Data*)(worker); |
18 } | |
19 | |
235 | 20 static void start_worker(Worker* worker) { |
21 CPUWorker* cpuWorker = (CPUWorker*)worker->worker; | |
22 cpuWorker->context = NEW(struct Context); | |
23 initContext(cpuWorker->context); | |
242 | 24 pthread_cond_wait(&cpuWorker->cond, &cpuWorker->mutex); |
235 | 25 goto meta(cpuWorker->context, C_taskReceiveWorker); |
233 | 26 } |
27 | |
244 | 28 __code taskReceiveWorker(struct Context* context, Worker* worker) { |
230 | 29 Queue* queue = worker->tasks; |
227 | 30 queue->next = C_getTask1; |
31 goto meta(context, queue->take); | |
32 } | |
33 | |
34 __code taskReceiveWorker_stub(struct Context* context) { | |
244 | 35 goto taskReceiveWorker(context, Gearef(context, Worker)); |
222
77faa28128b4
Add taskSend for TaskManager
Tatsuki IHA <e125716@ie.u-ryukyu.ac.jp>
parents:
217
diff
changeset
|
36 } |
77faa28128b4
Add taskSend for TaskManager
Tatsuki IHA <e125716@ie.u-ryukyu.ac.jp>
parents:
217
diff
changeset
|
37 |
230 | 38 __code getTask1(struct Context* context, Worker* worker, struct Context* task) { |
240 | 39 if (!task) |
236 | 40 return; // end thread |
230 | 41 task->worker = worker; |
42 goto meta(task, task->next); | |
169
ea7b11f3e717
Using Queue Interface
Tatsuki IHA <e125716@ie.u-ryukyu.ac.jp>
parents:
149
diff
changeset
|
43 } |
109
059b26a250cc
Change put_queue process
Tatsuki IHA <e125716@ie.u-ryukyu.ac.jp>
parents:
103
diff
changeset
|
44 |
169
ea7b11f3e717
Using Queue Interface
Tatsuki IHA <e125716@ie.u-ryukyu.ac.jp>
parents:
149
diff
changeset
|
45 __code getTask1_stub(struct Context* context) { |
230 | 46 Worker* worker = Gearef(context,Worker); |
244 | 47 Context* task = &worker->tasks->data->context; |
230 | 48 goto getTask1(context,worker,task); |
95
3e28ee215c0e
modify twice, use OSAtomiceCompareAndSwap
Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
49 } |
101 | 50 |
51 #ifdef USE_CUDA | |
98 | 52 __code twiceGpu(struct Context* context) { |
53 cuMemcpyHtoDAsync(context,context,context,context->stream); | |
54 cuLaunchkanel(); | |
55 cuMemcpyDtoHAsync(); | |
56 } | |
101 | 57 #endif |
227 | 58 |
232 | 59 __code shutdownWorker(struct Context* context, CPUWorker* worker) { |
230 | 60 } |
227 | 61 |
230 | 62 __code shutdownWorker_stub(struct Context* context) { |
232 | 63 CPUWorker* worker = (CPUWorker *)GearImpl(context, Worker, worker); |
230 | 64 goto shutdownWorker(context,worker); |
227 | 65 } |