loopy/schedule/__init__.py
def find_loop_nest_with_map(kernel):
    """Returns a dictionary mapping inames to other inames that are
    always nested with them.
    """
    result = {}

    from loopy.kernel.data import ParallelTag, IlpBaseTag, VectorizeTag

    all_nonpar_inames = set([
            iname
            for iname in kernel.all_inames()
            if not isinstance(kernel.iname_to_tag.get(iname),
                (ParallelTag, IlpBaseTag, VectorizeTag))])

    iname_to_insns = kernel.iname_to_insns()

    for iname in all_nonpar_inames:
        result[iname] = set([
            other_iname
            for insn in iname_to_insns[iname]
            for other_iname in kernel.insn_inames(insn) & all_nonpar_inames
            ])

    return result
Returns a dictionary mapping inames to other inames that are
    always nested with them.
    
Returns a dictionary mapping inames to other inames that are
always nested with them.
Returns a dictionary mapping inames to other inames that are always nested with them .
def find_loop_nest_with_map(kernel):
    result = {}
    from loopy.kernel.data import ParallelTag, IlpBaseTag, VectorizeTag
    all_nonpar_inames = set([
            iname
            for iname in kernel.all_inames()
            if not isinstance(kernel.iname_to_tag.get(iname),
                (ParallelTag, IlpBaseTag, VectorizeTag))])
    iname_to_insns = kernel.iname_to_insns()
    for iname in all_nonpar_inames:
        result[iname] = set([
            other_iname
            for insn in iname_to_insns[iname]
            for other_iname in kernel.insn_inames(insn) & all_nonpar_inames
            ])
    return result
def find_loop_nest_with_map ( kernel ) : result = { } from loopy . kernel . data import ParallelTag , IlpBaseTag , VectorizeTag all_nonpar_inames = set ( [ iname for iname in kernel . all_inames ( ) if not isinstance ( kernel . iname_to_tag . get ( iname ) , ( ParallelTag , IlpBaseTag , VectorizeTag ) ) ] ) iname_to_insns = kernel . iname_to_insns ( ) for iname in all_nonpar_inames : result [ iname ] = set ( [ other_iname for insn in iname_to_insns [ iname ] for other_iname in kernel . insn_inames ( insn ) & all_nonpar_inames ] ) return result
Returns a dictionary mapping inames to other inames that are
always nested with them.
Returns a dictionary mapping inames to other inames that are always nested with them .
"""Returns a dictionary mapping inames to other inames that are
    always nested with them.
    """
kernel