def kernel_name(op):
    """Returns kernel_name of a given cutlass op_instance."""
    from cutlass_lib import library

    threadblock = op.tile_description.procedural_name()
    extended_name = op.extended_name()
    opcode_class_name = library.OpcodeClassNames[
        op.tile_description.math_instruction.opcode_class
    ]
    layout = op.layout_name()
    align_ab = op.A.alignment
    align_c = op.C.alignment
    prefix = ""
    if op.prefix != "":
        kernel_schedule = library.KernelScheduleSuffixes[op.kernel_schedule]
        epilogue_schedule = library.EpilogueScheduleSuffixes[op.epilogue_schedule]
        prefix = f"{op.prefix}{kernel_schedule}{epilogue_schedule}"
    name = KERNEL_KEY_TEMPLATE.render(
        prefix=prefix,
        threadblock=threadblock,
        extended_name=extended_name,
        opcode_class_name=opcode_class_name,
        layout=layout,
        align_ab=align_ab,
        align_c=align_c,
    )
    return name.replace("\n", "")
Returns kernel_name of a given cutlass op_instance.
python/aitemplate/backend/cuda/gemm_universal/common.py