def METHOD_NAME(batch_size):
    benchmarker = ModuleBenchmarker(MLPNet())
    batches = [torch.rand(batch_size, 8) for _ in range(10)]
    warmup_size = 5
    test_size = 30
    results = benchmarker.run_batches_on_device(
        batches,
        "cpu",
        full_precision=True,
        warmup_size=warmup_size,
        test_size=test_size,
    )
    _results_sanity_check(results, test_size, batch_size)