Function fn

src/fused_mm_sampling/bench/speed_test.py:105–106

()

    sampler.prepare()

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

    return fn