def get_config(a_layout, b_layout, c_layout):
    """
    Return config function for given layouts of A, B, and bias.
    """

    def config(func_attrs, dtype="float16"):
        import cutlass_lib

        layout_choice = {
            "c": cutlass_lib.library.LayoutType.ColumnMajor,
            "r": cutlass_lib.library.LayoutType.RowMajor,
        }

        def fproc(op):
            return common.default_fproc(
                op=op,
                a_layout=layout_choice[a_layout],
                b_layout=layout_choice[b_layout],
                c_layout=layout_choice[c_layout],
                dtype=func_attrs["inputs"][0].dtype(),
                epilogue_name=func_attrs["epilogue"],
            )

        func_attrs["op_instance"] = common.extract_config(
            f_proc_op=fproc,
            include_cutlass_3x_ops=True,
        )

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