Function fn

src/fused_mm_sampling/bench/triton_benchmark_lib.py:174–175

()

    sampler.prepare()

    def fn():
        return sampler.sample(**kwargs)

    is_distributed = tp.size > 1
    match bench_fn: