diff --git a/benchmarks/run_sumpy_kernels.py b/benchmarks/run_sumpy_kernels.py index 898cd467f5766208addcb4b79a360a733353b2a9..92d29108cd149469a8cf0030c4940e33264c8e24 100644 --- a/benchmarks/run_sumpy_kernels.py +++ b/benchmarks/run_sumpy_kernels.py @@ -3,6 +3,7 @@ import numpy as np import pyopencl as cl import logging from dataclasses import dataclass +import time logger = logging.getLogger(__name__) @@ -110,6 +111,11 @@ class SumpyBenchmarkSuite: time_schedule.timeout = 600.0 time_generate_code.timeout = 600.0 + # Use CPU time as the timer + time_instantiate.timer = time.process_time + time_schedule.timer = time.process_time + time_generate_code.timer = time.process_time + # No warmup is needed time_instantiate.warmup_time = 0 time_schedule.warmup_time = 0