Commit 6568a69d authored by Paul's avatar Paul
Browse files

Seperate host/device time

parent 6fa85d83
...@@ -56,14 +56,22 @@ time_op(context& ictx, operation op, const std::vector<shape>& inputs, int n) ...@@ -56,14 +56,22 @@ time_op(context& ictx, operation op, const std::vector<shape>& inputs, int n)
op.compute(ctx, output, args); op.compute(ctx, output, args);
ctx.finish(); ctx.finish();
}; };
gctx.enable_perf_measurement();
run(); run();
// Measure host time
double host_time = 0.0; double host_time = 0.0;
double device_time = 0.0;
for(auto i : range(n)) for(auto i : range(n))
{ {
(void)i; (void)i;
host_time += time<milliseconds>(run); host_time += time<milliseconds>(run);
}
// Measure device time
gctx.enable_perf_measurement();
run();
double device_time = 0.0;
for(auto i : range(n))
{
(void)i;
run();
device_time += gctx.get_elapsed_ms(); device_time += gctx.get_elapsed_ms();
} }
return std::make_pair(host_time / n, device_time / n); return std::make_pair(host_time / n, device_time / n);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment