def has_tma_epilogue(op):
    """Check whether the op is CUTLASS 3.x and has a TMA epilogue schedule."""
    import cutlass_lib

    result = False
    if op.gemm_kind == cutlass_lib.library.GemmKind.Universal3x:
        epilogue_schedule_str = str(op.epilogue_schedule).split(".")[-1]
        result = epilogue_schedule_str.lower().startswith("tma")

    return result
Check whether the op is CUTLASS 3.x and has a TMA epilogue schedule.
python/aitemplate/backend/cuda/gemm_universal/common.py