def extract_config(op_kind, extra_kind):
    """
    Parameters
    ----------
    op_kind : ck_lib.library.OperationKind
        Operation kind.
    extra_kind : ck_lib.library.[AnyKind]
        Used to as extra flag to distinguish kernels.
        E.g. bias_add_relu vs. add_relu_bias

    Returns
    -------
    Dict
        Extracted (operation name, operation instance) pair
        from all operation candidates.
    """
    import ck_lib  # noqa: F401

    conv2d_ops = OrderedDict()
    extract_ops = list(Target.current()._operators[op_kind][extra_kind].items())
    for key, value in extract_ops:
        conv2d_ops[key] = value[0]
    return conv2d_ops
Parameters
----------
op_kind : ck_lib.library.OperationKind
Operation kind.
extra_kind : ck_lib.library.[AnyKind]
Used to as extra flag to distinguish kernels.
E.g. bias_add_relu vs. add_relu_bias

Returns
-------
Dict
Extracted (operation name, operation instance) pair
from all operation candidates.
python/aitemplate/backend/rocm/conv2d/common.py