Skip to content
Snippets Groups Projects
Commit 6ca0d8aa authored by Kaushik Kulkarni's avatar Kaushik Kulkarni Committed by Andreas Klöckner
Browse files

define :meth:`InKernelCallable.with_name`

parent e7c09088
No related branches found
No related tags found
1 merge request!426Discussion: kernel_callables_v3-edit2
......@@ -314,6 +314,7 @@ class InKernelCallable(ImmutableRecord):
.. automethod:: get_hw_axes_sizes
.. automethod:: get_used_hw_axes
.. automethod:: get_called_callables
.. automethod:: with_name
.. note::
......@@ -492,6 +493,13 @@ class InKernelCallable(ImmutableRecord):
"""
raise NotImplementedError
def with_name(self, name):
"""
Returns a copy of *self* so that it could be referred by *name*
in a :attr:`loopy.TranslationUnit.callables_table`'s namespace.
"""
raise NotImplementedError
# }}}
......@@ -655,6 +663,9 @@ class ScalarCallable(InKernelCallable):
"""
return frozenset()
def with_name(self, name):
return self
# }}}
......@@ -949,6 +960,10 @@ class CallableKernel(InKernelCallable):
return get_resolved_callable_ids_called_by_knl(self.subkernel,
callables_table)
def with_name(self, name):
new_knl = self.subkernel.copy(name=name)
return self.copy(subkernel=new_knl)
# }}}
......
......@@ -715,7 +715,7 @@ def add_callable_to_table(callables_table, clbl_id, clbl):
new_clbl_id = ung(clbl_id)
new_callables_table = callables_table.copy()
new_callables_table[new_clbl_id] = clbl
new_callables_table[new_clbl_id] = clbl.with_name(new_clbl_id)
return new_clbl_id, new_callables_table
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment