File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed
Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change 2929import islpy as isl
3030from islpy import dim_type
3131from pymbolic .mapper .stringifier import PREC_NONE
32+ from pytools import fset_union
3233
3334from loopy .codegen .control import build_loop_nest
3435from loopy .codegen .result import CodeGenerationResult , merge_codegen_results
@@ -294,8 +295,11 @@ def set_up_hw_parallel_loops(
294295 else :
295296 raise RuntimeError ("unexpected hw tag type" )
296297
298+ inames_in_subkernel = fset_union (
299+ kernel .id_to_insn [insn_id ].within_inames
300+ for insn_id in insn_ids_for_block )
297301 other_inames_with_same_tag = [
298- other_iname for other_iname in kernel . all_inames ()
302+ other_iname for other_iname in inames_in_subkernel
299303 if (kernel .iname_tags_of_type (other_iname , UniqueInameTag )
300304 and other_iname != iname
301305 and any (_tag .key == tag .key
You can’t perform that action at this time.
0 commit comments