Moving benchmark to another repository

pystencils/benchmark/__init__.py

-from .benchmark import generate_benchmark, kernel_header, kernel_source
-
-__all__ = [generate_benchmark, kernel_header, kernel_source]

pystencils/benchmark/benchmark.py

-from pathlib import Path
-
-from jinja2 import Environment, PackageLoader, StrictUndefined
-
-from pystencils.backends.cbackend import generate_c, get_headers
-from pystencils.astnodes import KernelFunction
-
-
-def generate_benchmark(kernel_ast: KernelFunction, path: Path = None, dialect='c', kernel_name: str = None):
-    if path is None:
-        path = Path('.')
-    if kernel_name is None:
-        kernel_name = kernel_ast.function_name
-
-    header = kernel_header(kernel_ast, dialect)
-    with open(path / f'{kernel_name}.h', 'w+') as f:
-        f.write(header)
-
-    source = kernel_source(kernel_ast, dialect)
-    with open(path / f'{kernel_name}.cpp', 'w+') as f:
-        f.write(source)
-
-
-def kernel_header(kernel_ast, dialect='c'):
-    function_signature = generate_c(kernel_ast, dialect=dialect, signature_only=True)
-
-    jinja_context = {
-        'function_signature': function_signature,
-    }
-
-    env = Environment(loader=PackageLoader('pystencils.benchmark'), undefined=StrictUndefined)
-    header = env.get_template('kernel.h').render(**jinja_context)
-    return header
-
-
-def kernel_source(kernel_ast: KernelFunction, dialect='c'):
-    kernel_name = kernel_ast.function_name
-    function_source = generate_c(kernel_ast, dialect=dialect)
-    headers = get_headers(kernel_ast)
-    headers.add(f'"{kernel_name}.h"')
-
-    jinja_context = {
-        'function_source': function_source,
-        'headers': sorted(headers),
-    }
-
-    env = Environment(loader=PackageLoader('pystencils.benchmark'), undefined=StrictUndefined)
-    header = env.get_template('kernel.c').render(**jinja_context)
-    return header

pystencils/benchmark/templates/kernel.c

-{% for header in headers %}#include {{header}}
-{% endfor %}
-
-#define RESTRICT __restrict__
-#define FUNC_PREFIX
-
-{{function_source}}

pystencils/benchmark/templates/kernel.h

-#define RESTRICT __restrict__
-#define FUNC_PREFIX
-
-{{function_signature}};