Function run_own_benchmark

src/fused_mm_sampling/bench/speed_test.py:250–259

(args: Args)

def run_own_benchmark(args: Args) -> None:
    tp = args.make_tp()
    cases: list[Case] = all_cases(args)
    if args.nsys_profile:
        for case in cases:
            nsys_profile(case)
    else:
        df = benchmark_all(cases)
        if tp.is_rank0():
            _print_and_dump_own_results(df, args)


def _print_and_dump_own_results(df: pd.DataFrame, args: Args) -> None: