def gen_kernels()

in gen_kernels.py [0:0]

• 13 lines of code
• 9 McCabe index (conditional complexity)

def gen_kernels():
    for prefix in ['s', 'h']:
        for op in ['NN', 'NT', 'TN', 'TT']:
            for tileM, tileN, tileK, vecA, vecB, vecC, div, base_op, dyn_shared in selections[prefix][op]:
                for vec in [False, True]:
                    for major, minor in [(5, 0), (6, 0)]:
                        if base_op:
                            # The op is part of the base kernel name
                            base = "%sgemm_%dx%dx%d_%s" % (prefix, tileM, tileN, tileK, op)
                            opts = ( "vec", ) if vec else ()
                        else:
                            # The op is an option passed to a more generic kernel
                            base = "%sgemm_%dx%dx%d" % (prefix, tileM, tileN, tileK)
                            opts = ( op, "vec" ) if vec else (op,)

                        get_kernel(base, major, minor, opts)