| mtklein@google.com | d36522d | 2013-10-16 13:02:15 +0000 | [diff] [blame] | 1 | #include "DMTaskRunner.h" |
| 2 | #include "DMTask.h" |
| 3 | |
| 4 | namespace DM { |
| 5 | |
| 6 | TaskRunner::TaskRunner(int cputhreads, int gpuThreads) |
| 7 | : fMain(cputhreads) |
| 8 | , fGpu(gpuThreads) |
| 9 | {} |
| 10 | |
| 11 | void TaskRunner::add(Task* task) { |
| 12 | if (task->usesGpu()) { |
| 13 | fGpu.add(task); |
| 14 | } else { |
| 15 | fMain.add(task); |
| 16 | } |
| 17 | } |
| 18 | |
| 19 | void TaskRunner::wait() { |
| 20 | // These wait calls block until the threadpool is done. We don't allow |
| 21 | // children to spawn new GPU tasks so we can wait for that first knowing |
| 22 | // we'll never try to add to it later. Same can't be said of fMain: fGpu |
| 23 | // and fMain can both add tasks to fMain, so we have to wait for that last. |
| 24 | fGpu.wait(); |
| 25 | fMain.wait(); |
| 26 | } |
| 27 | |
| 28 | } // namespace DM |