Index: dm/DM.cpp |
diff --git a/dm/DM.cpp b/dm/DM.cpp |
index b330445186444d7fd6d93292013ad1a7b2c85691..09f0ce8f268cd0b15e0cd16a3012fd4e19148741 100644 |
--- a/dm/DM.cpp |
+++ b/dm/DM.cpp |
@@ -8,7 +8,6 @@ |
#include "SkGraphics.h" |
#include "SkPicture.h" |
#include "SkString.h" |
-#include "SkTaskGroup.h" |
#include "Test.h" |
#include "gm.h" |
#include "sk_tool_utils.h" |
@@ -42,6 +41,7 @@ |
static const char kGpuAPINameGL[] = "gl"; |
static const char kGpuAPINameGLES[] = "gles"; |
+DEFINE_int32(gpuThreads, 1, "Threads for GPU work."); |
DEFINE_string2(expectations, r, "", |
"If a directory, compare generated images against images under this path. " |
"If a file, compare generated images against JSON expectations at this path." |
@@ -192,7 +192,6 @@ |
int dm_main() { |
SetupCrashHandler(); |
SkAutoGraphics ag; |
- SkTaskGroup::Enabler enabled(FLAGS_threads); |
if (FLAGS_dryRun) { |
FLAGS_verbose = true; |
@@ -234,8 +233,7 @@ |
SkDebugf("%d GMs x %d configs, %d tests, %d pictures\n", |
gms.count(), configs.count(), tests.count(), skps.count()); |
DM::Reporter reporter; |
- |
- DM::TaskRunner tasks; |
+ DM::TaskRunner tasks(FLAGS_threads, FLAGS_gpuThreads); |
kick_off_tests(tests, &reporter, &tasks); |
kick_off_gms(gms, configs, gpuAPI, *expectations, &reporter, &tasks); |
kick_off_skps(skps, &reporter, &tasks); |