diff --git a/loopy/__init__.py b/loopy/__init__.py index 24d50d8448b609c21bfc6732ef9ac939d34f7fd3..3027e10d38a98b4c86737502b7fd20f41c3cc3c3 100644 --- a/loopy/__init__.py +++ b/loopy/__init__.py @@ -1032,7 +1032,7 @@ def add_prefetch(kernel, var_name, sweep_inames=[], dim_arg_names=None, def find_instructions(kernel, insn_match): from loopy.context_matching import parse_id_match match = parse_id_match(insn_match) - return [insn for insn in kernel.instructions if match(insn.id, None)] + return [insn for insn in kernel.instructions if match(insn.id, insn.tags)] def map_instructions(kernel, insn_match, f):