[harness] Fix runner - pass device to timer

Author: adam-smnk

ai_bench/harness/runner/kernel_bench_runner.py

@@ -102,5 +102,7 @@ def run_kernels(self):
                         continue
 
                     print(f"Benchmarking: {variant}")
-                    meas = testing.time(fn, args, warmup=self.warmup, rep=self.rep)
+                    meas = testing.time(
+                        fn, args, warmup=self.warmup, rep=self.rep, device=self.device
+                    )
                     print(f"time: {meas}us")