Skip to content
Snippets Groups Projects
Commit a9ddc5b6 authored by Julian Hammer's avatar Julian Hammer
Browse files

nicer interface with (optional) preservation of temp files

parent 2922cdba
No related merge requests found
...@@ -23,8 +23,8 @@ class PyStencilsKerncraftKernel(kerncraft.kernel.KernelCode): ...@@ -23,8 +23,8 @@ class PyStencilsKerncraftKernel(kerncraft.kernel.KernelCode):
""" """
LIKWID_BASE = '/usr/local/likwid' LIKWID_BASE = '/usr/local/likwid'
def __init__(self, ast: KernelFunction, machine: Optional[MachineModel] = None, assumed_layout='SoA', def __init__(self, ast: KernelFunction, machine: Optional[MachineModel] = None,
debug_print=False): assumed_layout='SoA', debug_print=False, filename=None):
"""Create a kerncraft kernel using a pystencils AST """Create a kerncraft kernel using a pystencils AST
Args: Args:
...@@ -38,7 +38,7 @@ class PyStencilsKerncraftKernel(kerncraft.kernel.KernelCode): ...@@ -38,7 +38,7 @@ class PyStencilsKerncraftKernel(kerncraft.kernel.KernelCode):
# Initialize state # Initialize state
self.asm_block = None self.asm_block = None
self._filename = None self._filename = filename
self.kernel_ast = ast self.kernel_ast = ast
self.temporary_dir = TemporaryDirectory() self.temporary_dir = TemporaryDirectory()
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment