def _get_op_checker(op_type: str) -> Callable:
    """Returns op specific check functions to check if two ops can be
    grouped togher.
    """
    if op_type.startswith("gemm"):
        return _gemm_op_check
    if op_type.startswith("layernorm"):
        return _layernorm_op_check
    return _default_op_check
Returns op specific check functions to check if two ops can be
grouped togher.
python/aitemplate/compiler/transform/fuse_group_ops.py