def generate_single_cell_wrapper(iterset, args, forward_args=(), kernel_name=None, wrapper_name=None):
"""Generates wrapper for a single cell. No iteration loop, but cellwise data is extracted.
Cell is expected as an argument to the wrapper. For extruded, the numbering of the cells
is columnwise continuous, bottom to top.
:param iterset: The iteration set
:param args: :class:`Arg`s
:param forward_args: To forward unprocessed arguments to the kernel via the wrapper,
give an iterable of strings describing their C types.
:param kernel_name: Kernel function name
:param wrapper_name: Wrapper function name
:return: string containing the C code for the single-cell wrapper
"""
from pyop2.codegen.builder import WrapperBuilder
from pyop2.codegen.rep2loopy import generate
from loopy.types import OpaqueType
forward_arg_types = [OpaqueType(fa) for fa in forward_args]
builder = WrapperBuilder(iterset=iterset, single_cell=True, forward_arg_types=forward_arg_types)
for arg in args:
builder.add_argument(arg)
builder.set_kernel(Kernel("", kernel_name))
wrapper = generate(builder, wrapper_name)
code = loopy.generate_code_v2(wrapper)
return code.device_code()
def generate_single_cell_wrapper(iterset,
args,
forward_args=(),
kernel_name=None,
wrapper_name=None):
"""Generates wrapper for a single cell. No iteration loop, but cellwise data is extracted.
Cell is expected as an argument to the wrapper. For extruded, the numbering of the cells
is columnwise continuous, bottom to top.
:param iterset: The iteration set
:param args: :class:`Arg`s
:param forward_args: To forward unprocessed arguments to the kernel via the wrapper,
give an iterable of strings describing their C types.
:param kernel_name: Kernel function name
:param wrapper_name: Wrapper function name
:return: string containing the C code for the single-cell wrapper
"""
from pyop2.codegen.builder import WrapperBuilder
from pyop2.codegen.rep2loopy import generate
from loopy.types import OpaqueType
forward_arg_types = [OpaqueType(fa) for fa in forward_args]
builder = WrapperBuilder(iterset=iterset,
single_cell=True,
forward_arg_types=forward_arg_types)
for arg in args:
builder.add_argument(arg)
builder.set_kernel(Kernel("", kernel_name))
wrapper = generate(builder, wrapper_name)
code = loopy.generate_code_v2(wrapper)
return code.device_code()
def code_to_compile(self):
from pyop2.codegen.builder import WrapperBuilder
from pyop2.codegen.rep2loopy import generate
builder = WrapperBuilder(iterset=self._iterset, iteration_region=self._iteration_region, pass_layer_to_kernel=self._pass_layer_arg)
for arg in self._args:
builder.add_argument(arg)
builder.set_kernel(self._kernel)
wrapper = generate(builder)
code = loopy.generate_code_v2(wrapper)
if self._kernel._cpp:
from loopy.codegen.result import process_preambles
preamble = "".join(process_preambles(getattr(code, "device_preambles", [])))
device_code = "\n\n".join(str(dp.ast) for dp in code.device_programs)
return preamble + "\nextern \"C\" {\n" + device_code + "\n}\n"
return code.device_code()
def code_to_compile(self):
from pyop2.codegen.builder import WrapperBuilder
from pyop2.codegen.rep2loopy import generate
builder = WrapperBuilder(kernel=self._kernel,
iterset=self._iterset,
iteration_region=self._iteration_region,
pass_layer_to_kernel=self._pass_layer_arg)
for arg in self._args:
builder.add_argument(arg)
wrapper = generate(builder)
code = loopy.generate_code_v2(wrapper)
if self._kernel._cpp:
from loopy.codegen.result import process_preambles
preamble = "".join(process_preambles(getattr(code, "device_preambles", [])))
device_code = "\n\n".join(str(dp.ast) for dp in code.device_programs)
return preamble + "\nextern \"C\" {\n" + device_code + "\n}\n"
return code.device_code()