def get_gen_profiler(a_layout, b_layout, c_layout):
    """
    Return gen_profiler for given layouts of A, B, and bias.
    """

    def gen_profiler(func_attrs, workdir, profiler_filename, dim_info_dict):
        problem_args = _get_problem_args(a_layout, b_layout, c_layout)
        return bmm_common.default_gen_profiler(
            func_attrs,
            workdir,
            profiler_filename,
            dim_info_dict,
            problem_args,
        )

    return gen_profiler
Return gen_profiler for given layouts of A, B, and bias.
python/aitemplate/backend/cuda/gemm_universal/bmm_xxx.py