Function run_speed_test

src/fused_mm_sampling/bench/speed_test.py:279–281

Run a speed test for a given set of arguments.

(args: Args)

def run_speed_test(args: Args) -> None:
    """Run a speed test for a given set of arguments."""
    run_maybe_distributed(_run_speed_test_impl, args.n_procs, args)


def _run_speed_test_impl(args: Args) -> None: