Skip to content
Snippets Groups Projects
Commit 1033e2b0 authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Add remove_unused_arguments()

parent 5b0739d4
No related branches found
Tags v2016.1.1
No related merge requests found
......@@ -430,6 +430,8 @@ Influencing data access
.. autofunction:: tag_data_axes
.. autofunction:: remove_unused_arguments
Padding
^^^^^^^
......
......@@ -2090,4 +2090,22 @@ def to_batched(knl, nbatches, batch_varying_args, batch_iname_prefix="ibatch"):
# }}}
# {{{ remove_unused_arguments
def remove_unused_arguments(knl):
new_args = []
refd_vars = set(knl.all_params())
for insn in knl.instructions:
refd_vars.update(insn.dependency_names())
for arg in knl.args:
if arg.name in refd_vars:
new_args.append(arg)
return knl.copy(args=new_args)
# }}}
# vim: foldmethod=marker
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