add flexible input options for smem_args

kernel_tuner/core.py

@@ -366,7 +366,7 @@ def compile_and_benchmark(self, kernel_source, gpu_args, params, kernel_options,
 
             #add shared memory arguments to compiled module
             if kernel_options.smem_args is not None:
-                self.dev.copy_shared_memory_args(kernel_options.smem_args)
+                self.dev.copy_shared_memory_args(util.get_smem_args(kernel_options.smem_args, params))
             #add constant memory arguments to compiled module
             if kernel_options.cmem_args is not None:
                 self.dev.copy_constant_memory_args(kernel_options.cmem_args)

kernel_tuner/interface.py

@@ -552,7 +552,7 @@ def run_kernel(kernel_name, kernel_string, problem_size, arguments, params, grid
 
             #add shared memory arguments to compiled module
             if smem_args is not None:
-                dev.copy_shared_memory_args(smem_args)
+                dev.copy_shared_memory_args(util.get_smem_args(smem_args, params))
             #add constant memory arguments to compiled module
             if cmem_args is not None:
                 dev.copy_constant_memory_args(cmem_args)

kernel_tuner/util.py

@@ -304,6 +304,19 @@ def get_problem_size(problem_size, params):
     return current_problem_size
 
 
+def get_smem_args(smem_args, params):
+    """ return a dict with kernel instance specific size """
+    result = smem_args.copy()
+    if 'size' in result:
+        size = result['size']
+        if callable(size):
+            size = size(params)
+        elif isinstance(size, str):
+            size = util.replace_param_occurrences(size, params)
+            size = int(eval(size))
+    return result
+
+
 def get_temp_filename(suffix=None):
     """ return a string in the form of temp_X, where X is a large integer """
     file = tempfile.mkstemp(