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 | |
commit-bot@chromium.org | ef57b7e | 2014-02-28 20:31:31 +0000 | [diff] [blame] | 6 | TaskRunner::TaskRunner(int cpuThreads, int gpuThreads) : fCpu(cpuThreads), fGpu(gpuThreads) {} |
commit-bot@chromium.org | 0dc5bd1 | 2014-02-26 16:31:22 +0000 | [diff] [blame] | 7 | |
commit-bot@chromium.org | ef57b7e | 2014-02-28 20:31:31 +0000 | [diff] [blame] | 8 | void TaskRunner::add(CpuTask* task) { fCpu.add(task); } |
commit-bot@chromium.org | 3f03215 | 2014-05-01 17:41:32 +0000 | [diff] [blame] | 9 | void TaskRunner::addNext(CpuTask* task) { fCpu.addNext(task); } |
commit-bot@chromium.org | ef57b7e | 2014-02-28 20:31:31 +0000 | [diff] [blame] | 10 | void TaskRunner::add(GpuTask* task) { fGpu.add(task); } |
mtklein@google.com | d36522d | 2013-10-16 13:02:15 +0000 | [diff] [blame] | 11 | |
| 12 | void TaskRunner::wait() { |
commit-bot@chromium.org | ef57b7e | 2014-02-28 20:31:31 +0000 | [diff] [blame] | 13 | // These wait calls block until each threadpool is done. We don't allow |
| 14 | // spawning new child GPU tasks, so we can wait for that first knowing |
| 15 | // we'll never try to add to it later. Same can't be said of the CPU pool: |
| 16 | // both CPU and GPU tasks can spawn off new CPU work, so we wait for that last. |
mtklein@google.com | d36522d | 2013-10-16 13:02:15 +0000 | [diff] [blame] | 17 | fGpu.wait(); |
commit-bot@chromium.org | ef57b7e | 2014-02-28 20:31:31 +0000 | [diff] [blame] | 18 | fCpu.wait(); |
mtklein@google.com | d36522d | 2013-10-16 13:02:15 +0000 | [diff] [blame] | 19 | } |
| 20 | |
| 21 | } // namespace DM |