def get_default_problem_info(default_problem_args, **kwargs):
    """Return the default problem args"""
    problem_args = default_problem_args.copy()
    for k, v in kwargs.items():
        problem_args[k] = v

    bmm_problem_info = Bmm_problem_info(**problem_args)
    return bmm_problem_info
Return the default problem args
python/aitemplate/backend/cuda/gemm_universal/bmm_common.py