| Index: dm/DMUtil.h | 
| diff --git a/dm/DMUtil.h b/dm/DMUtil.h | 
| index a79164946ea56068f793e6ac4466ebbfdb342175..faa4e39737fe8af007e28da726a519577ac52247 100644 | 
| --- a/dm/DMUtil.h | 
| +++ b/dm/DMUtil.h | 
| @@ -1,6 +1,7 @@ | 
| #ifndef DMUtil_DEFINED | 
| #define DMUtil_DEFINED | 
|  | 
| +#include "SkBenchmark.h" | 
| #include "SkBitmap.h" | 
| #include "SkString.h" | 
| #include "gm_expectations.h" | 
| @@ -15,8 +16,10 @@ SkString UnderJoin(const char* a, const char* b); | 
| // Draw gm to picture.  Passes recordFlags to SkPicture::beginRecording(). | 
| void RecordPicture(skiagm::GM* gm, SkPicture* picture, uint32_t recordFlags = 0); | 
|  | 
| -// Prepare bitmap to have gm draw into it with this config. | 
| +// Prepare bitmap to have gm or bench draw into it with this config. | 
| +// TODO(mtklein): make SkBenchmark::getSize()/GM::getISize() const. | 
| void SetupBitmap(const SkColorType, skiagm::GM* gm, SkBitmap* bitmap); | 
| +void SetupBitmap(const SkColorType, SkBenchmark* bench, SkBitmap* bitmap); | 
|  | 
| // Draw picture to bitmap. | 
| void DrawPicture(SkPicture* picture, SkBitmap* bitmap); | 
|  |