Index: tools/VisualBench/VisualBenchmarkStream.cpp |
diff --git a/tools/VisualBench/VisualBenchmarkStream.cpp b/tools/VisualBench/VisualBenchmarkStream.cpp |
index 4cc819ded5e0f252ae2df288f04023f04ffec4e8..9e1ce36fbc12ec189a101ba832b3ee0288c7c76f 100644 |
--- a/tools/VisualBench/VisualBenchmarkStream.cpp |
+++ b/tools/VisualBench/VisualBenchmarkStream.cpp |
@@ -7,12 +7,14 @@ |
*/ |
#include <VisualBench/VisualBenchmarkStream.h> |
+#include "CpuWrappedBenchmark.h" |
#include "GMBench.h" |
#include "SkOSFile.h" |
#include "SkPictureRecorder.h" |
#include "SkStream.h" |
#include "VisualSKPBench.h" |
+DEFINE_bool(cpu, false, "Run in CPU mode?"); |
DEFINE_string2(match, m, nullptr, |
"[~][^]substring[$] [...] of bench name to run.\n" |
"Multiple matches may be separated by spaces.\n" |
@@ -72,6 +74,9 @@ Benchmark* VisualBenchmarkStream::next() { |
!bench->isSuitableFor(Benchmark::kGPU_Backend))) { |
bench->unref(); |
} |
+ if (FLAGS_cpu) { |
+ return new CpuWrappedBenchmark(bench); |
+ } |
return bench; |
} |