diff --git a/src/pystencils_autodiff/backends/python_bindings.py b/src/pystencils_autodiff/backends/python_bindings.py index d86c4f1c1dfa8738a22518ff8462497ee8f0f77e..e2f56f8209e9026e8c19cec21f9114548f7930d9 100644 --- a/src/pystencils_autodiff/backends/python_bindings.py +++ b/src/pystencils_autodiff/backends/python_bindings.py @@ -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] }) diff --git a/src/pystencils_autodiff/framework_integration/printer.py b/src/pystencils_autodiff/framework_integration/printer.py index 6e11cf684a67651fd1e45c92638d7321f80002cc..922535228d705ca4088bd36ae0bd26dc7f13025b 100644 --- a/src/pystencils_autodiff/framework_integration/printer.py +++ b/src/pystencils_autodiff/framework_integration/printer.py @@ -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