| Index: cc/raster/task_graph_runner.h | 
| diff --git a/cc/raster/task_graph_runner.h b/cc/raster/task_graph_runner.h | 
| index 23094ef5a7ef3c81a6cae0fbde931c57c11efe98..aa36526e5ed475d5562314e750d1dd805c65f4c8 100644 | 
| --- a/cc/raster/task_graph_runner.h | 
| +++ b/cc/raster/task_graph_runner.h | 
| @@ -15,7 +15,10 @@ | 
| #include "base/logging.h" | 
| #include "base/memory/ref_counted.h" | 
| #include "base/memory/scoped_ptr.h" | 
| +#include "base/threading/simple_thread.h" | 
| +#include "base/time/time.h" | 
| #include "cc/base/cc_export.h" | 
| +#include "cc/raster/task_category.h" | 
|  | 
| namespace cc { | 
|  | 
| @@ -33,6 +36,11 @@ class CC_EXPORT Task : public base::RefCountedThreadSafe<Task> { | 
| // safety. | 
| virtual void RunOnWorkerThread() = 0; | 
|  | 
| +  virtual bool IsRasterTask() { return false; } | 
| + | 
| +  void SetRunner(base::TestSimpleThread* runner) { runner_ = runner; } | 
| +  base::TestSimpleThread* GetRunner() { return runner_; } | 
| + | 
| void WillRun(); | 
| void DidRun(); | 
| bool HasFinishedRunning() const; | 
| @@ -45,6 +53,8 @@ class CC_EXPORT Task : public base::RefCountedThreadSafe<Task> { | 
|  | 
| bool will_run_; | 
| bool did_run_; | 
| + | 
| +  base::TestSimpleThread* runner_; | 
| }; | 
|  | 
| // A task dependency graph describes the order in which to execute a set | 
|  |