Skip to content
Snippets Groups Projects
Commit 41167d51 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

DIRTY HACK!!

parent 513acdfe
Branches
Tags
No related merge requests found
......@@ -170,5 +170,6 @@ class PybindFunctionWrapping(JinjaCppFile):
def __init__(self, function_node):
super().__init__({'python_name': function_node.function_name,
'cpp_name': function_node.function_name,
'parameters': [p.symbol.name for p in function_node.get_parameters()]
'parameters': [p.symbol.name for p in function_node.get_parameters()
if hasattr(p.symbol, 'dtype') and not 'meshFunctor' in p.symbol.name]
})
......@@ -58,7 +58,8 @@ class FrameworkIntegrationPrinter(pystencils.backends.cbackend.CBackend):
max_threads = node.indexing.max_threads_per_block()
if max_threads:
launch_bounds = f"__launch_bounds__({max_threads}) "
func_declaration = f"FUNC_PREFIX {launch_bounds} {self._print(return_type)} {node.function_name}({', '.join(function_arguments)})" # noqa
indent = ',\n' + ' ' * len(f'FUNC_PREFIX {launch_bounds} {self._print(return_type)} {node.function_name}')
func_declaration = f"FUNC_PREFIX {launch_bounds} {self._print(return_type)} {node.function_name}({indent.join(function_arguments)})" # noqa
if self._signatureOnly:
return func_declaration
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment