Return kernelMs from benchmarks demo. (#722)

Nikhil Thorat · web-flow · commit 56a3325479ce · 2018-02-14T15:19:19.000-05:00
diff --git a/demos/benchmarks/benchmark_util.ts b/demos/benchmarks/benchmark_util.ts
@@ -7,6 +7,5 @@ export async function warmupAndBenchmarkGPU(benchmark: () => dl.Tensor):
   const out = benchmark();
   await out.data();
   out.dispose();
-  // Real timing.
-  return dl.time(benchmark);
+  return (await dl.time(benchmark)).kernelMs;
 }

Original file line number	Diff line number	Diff line change
`@@ -7,6 +7,5 @@ export async function warmupAndBenchmarkGPU(benchmark: () => dl.Tensor):`
`7`	`7`	`const out = benchmark();`
`8`	`8`	`await out.data();`
`9`	`9`	`out.dispose();`
`10`		`- // Real timing.`
`11`		`- return dl.time(benchmark);`
	`10`	`+ return (await dl.time(benchmark)).kernelMs;`
`12`	`11`	`}`