Have a base class for runners.

Author: isazi

kernel_tuner/runners/runner.py

@@ -0,0 +1,18 @@
+"""This module contains the interface for runners."""
+from __future__ import print_function
+
+from abc import ABC, abstractmethod
+
+class Runner(ABC):
+    """Base class for kernel_tuner runners"""
+    @abstractmethod
+    def __init__(self, kernel_source, kernel_options, device_options, iterations, observers):
+        pass
+
+    @abstractmethod
+    def get_environment(self):
+        pass
+
+    @abstractmethod
+    def run(self, parameter_space, tuning_options):
+        pass

kernel_tuner/runners/sequential.py

@@ -7,9 +7,10 @@
 from kernel_tuner.core import DeviceInterface
 from kernel_tuner.util import (ErrorConfig, print_config_output,
                                process_metrics, store_cache)
+from kernel_tuner.runners.runner import Runner
 
 
-class SequentialRunner:
+class SequentialRunner(Runner):
     """ SequentialRunner is used for tuning with a single process/thread """
 
     def __init__(self, kernel_source, kernel_options, device_options, iterations, observers):

kernel_tuner/runners/simulation.py

@@ -4,6 +4,7 @@
 from time import perf_counter
 
 from kernel_tuner import util
+from kernel_tuner.runners.runner import Runner
 
 _SimulationDevice = namedtuple("_SimulationDevice", ["max_threads", "env", "quiet"])
 
@@ -25,7 +26,7 @@ def get_environment(self):
         return self.env
 
 
-class SimulationRunner:
+class SimulationRunner(Runner):
     """ SimulationRunner is used for tuning with a single process/thread """
 
     def __init__(self, kernel_source, kernel_options, device_options, iterations, observers):