Skip to content
Snippets Groups Projects

Kernel wrapper

Merged Stephan Seitz requested to merge seitz/pystencils:KernelWrapper into master
5 files
+ 30
13
Compare changes
  • Side-by-side
  • Inline
Files
5
+ 1
10
@@ -60,6 +60,7 @@ from appdirs import user_cache_dir, user_config_dir
@@ -60,6 +60,7 @@ from appdirs import user_cache_dir, user_config_dir
from pystencils import FieldType
from pystencils import FieldType
from pystencils.backends.cbackend import generate_c, get_headers
from pystencils.backends.cbackend import generate_c, get_headers
from pystencils.include import get_pystencils_include_path
from pystencils.include import get_pystencils_include_path
 
from pystencils.kernel_wrapper import KernelWrapper
from pystencils.utils import atomic_file_write, file_handle_for_atomic_write, recursive_dict_update
from pystencils.utils import atomic_file_write, file_handle_for_atomic_write, recursive_dict_update
@@ -482,16 +483,6 @@ class ExtensionModuleCode:
@@ -482,16 +483,6 @@ class ExtensionModuleCode:
print(create_module_boilerplate_code(self.module_name, self._function_names), file=file)
print(create_module_boilerplate_code(self.module_name, self._function_names), file=file)
class KernelWrapper:
def __init__(self, kernel, parameters, ast_node):
self.kernel = kernel
self.parameters = parameters
self.ast = ast_node
def __call__(self, **kwargs):
return self.kernel(**kwargs)
def compile_module(code, code_hash, base_dir):
def compile_module(code, code_hash, base_dir):
compiler_config = get_compiler_config()
compiler_config = get_compiler_config()
extra_flags = ['-I' + get_paths()['include'], '-I' + get_pystencils_include_path()]
extra_flags = ['-I' + get_paths()['include'], '-I' + get_pystencils_include_path()]
Loading