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

Accept more iterables for sweep_inames in add_prefetch()

parent 56ab4189
No related branches found
No related tags found
No related merge requests found
......@@ -1127,6 +1127,9 @@ def add_prefetch(kernel, var_name, sweep_inames=[], dim_arg_names=None,
if isinstance(sweep_inames, str):
sweep_inames = [s.strip() for s in sweep_inames.split(",")]
else:
# copy, standardize to list
sweep_inames = list(sweep_inames)
kernel, subst_use, sweep_inames, inames_to_be_removed = \
_process_footprint_subscripts(
......
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