def gen_function_decl(func_name, gemm_flag, pdims=0):
    """Generates function declarations.

    Parameters
    ----------
    func_attrs : Dict
        Operation attributes.
    gemm_flag : str
        Flag telling which backend should be generated. options are '','bias','bias_relu'.

    Returns
    -------
    str
        The rentered template of function declaration.
    """
    return common.gen_function_decl(
        func_name=func_name, gemm_flag=gemm_flag, ndims=3, pdims=pdims
    )
Generates function declarations.

    Parameters
    ----------
    func_attrs : Dict
        Operation attributes.
    gemm_flag : str
        Flag telling which backend should be generated. options are '','bias','bias_relu'.

    Returns
    -------
    str
        The rentered template of function declaration.
    
python/aitemplate/backend/rocm/gemm/bmm_common.py