| Index: tools/VisualBench/VisualBenchmarkStream.cpp
|
| diff --git a/tools/VisualBench/VisualBenchmarkStream.cpp b/tools/VisualBench/VisualBenchmarkStream.cpp
|
| index 235200a761295a69e9e8f84478eba33a5941e25f..906aab27886d44d9fbbaaf6bb6954595b3c33e6d 100644
|
| --- a/tools/VisualBench/VisualBenchmarkStream.cpp
|
| +++ b/tools/VisualBench/VisualBenchmarkStream.cpp
|
| @@ -7,13 +7,14 @@
|
| */
|
|
|
| #include <VisualBench/VisualBenchmarkStream.h>
|
| -#include "CpuWrappedBenchmark.h"
|
| +#include <VisualBench/WrappedBenchmark.h>
|
| #include "GMBench.h"
|
| #include "SkOSFile.h"
|
| #include "SkPath.h"
|
| #include "SkPictureRecorder.h"
|
| #include "SkStream.h"
|
| #include "sk_tool_utils.h"
|
| +#include "VisualFlags.h"
|
| #include "VisualSKPBench.h"
|
|
|
| DEFINE_bool(cpu, false, "Run in CPU mode?");
|
| @@ -115,8 +116,12 @@ Benchmark* VisualBenchmarkStream::next() {
|
| bench->unref();
|
| }
|
| }
|
| +
|
| + // TODO move this all to --config
|
| if (bench && FLAGS_cpu) {
|
| bench = new CpuWrappedBenchmark(bench);
|
| + } else if (bench && FLAGS_nvpr) {
|
| + bench = new NvprWrappedBenchmark(bench, 4);
|
| }
|
|
|
| fBenchmark.reset(bench);
|
|
|